fixup! fix(base_quantity): Removed string return from serialization o…

…f pint quantity.
NREL · Jun 5, 2024 · 26521e1 · 26521e1
1 parent 4f03c6f
commit 26521e1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/infrasys/base_quantity.py b/src/infrasys/base_quantity.py
@@ -68,7 +68,7 @@ def _serialize(cls, input_value, info: SerializationInfo) -> float | str:
             # We can add more logic that will change the serialization here.
             magnitude_only = context.get("magnitude_only")
             if magnitude_only:
-                return_value = return_value.magnitude
+                return return_value.magnitude
         if info.mode == "json":
             return_value = str(return_value)
         return return_value

diff --git a/tests/test_base_quantity.py b/tests/test_base_quantity.py
@@ -100,4 +100,4 @@ def test_custom_serialization():
     assert model_dump["voltage"] == 10.0
 
     model_dump = component.model_dump(mode="json", context={"magnitude_only": True})
-    assert model_dump["voltage"] == "10.0"
+    assert model_dump["voltage"] == 10.0