Skip to content

Commit bb44ad2

Browse files
Andrzej KrupkaAndrzej Krupka
Andrzej Krupka
authored and
Andrzej Krupka
committed
Fix unit tests after alias changes
1 parent 9242e4c commit bb44ad2

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

flow360/component/simulation/services.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
from typing import Any, Collection, Dict, Literal, Optional, Tuple, Union
99

1010
import pydantic as pd
11-
1211
from unyt import unyt_array, unyt_quantity
1312
from unyt.exceptions import UnitParseError
1413

flow360/component/simulation/user_code.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from pydantic import BeforeValidator
99
from typing_extensions import Self
1010
from unyt import Unit, unyt_array
11+
1112
from flow360.component.simulation.blueprint import Evaluable, expr_to_model
1213
from flow360.component.simulation.blueprint.codegen import expr_to_code
1314
from flow360.component.simulation.blueprint.core import EvaluationContext
@@ -283,8 +284,9 @@ def _validate_expression(cls, value) -> Self:
283284

284285
return {"expression": expression}
285286

286-
287-
def evaluate(self, context: EvaluationContext = None, strict: bool = True) -> Union[float, list[float], unyt_array]:
287+
def evaluate(
288+
self, context: EvaluationContext = None, strict: bool = True
289+
) -> Union[float, list[float], unyt_array]:
288290
if context is None:
289291
context = _global_ctx
290292
expr = expr_to_model(self.expression, context)
@@ -461,7 +463,7 @@ def _deserialize(value) -> Self:
461463

462464
def _serializer(value, info) -> dict:
463465
if isinstance(value, Expression):
464-
serialized = SerializedValueOrExpression(typeName="expression")
466+
serialized = SerializedValueOrExpression(type_name="expression")
465467

466468
serialized.expression = value.expression
467469

@@ -478,7 +480,7 @@ def _serializer(value, info) -> dict:
478480

479481
serialized.evaluated_units = str(evaluated.units.expr)
480482
else:
481-
serialized = SerializedValueOrExpression(typeName="number")
483+
serialized = SerializedValueOrExpression(type_name="number")
482484
if isinstance(value, Number):
483485
serialized.value = value
484486
elif isinstance(value, unyt_array):

0 commit comments

Comments
 (0)