Skip to content

Commit c35e081

Browse files
chore: type
Signed-off-by: jaapschoutenalliander <[email protected]>
1 parent b1da441 commit c35e081

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/power_grid_model_ds/_core/utils/serialization.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,9 @@ def _restore_grid_values(grid, json_data: Dict) -> None:
3131
continue
3232

3333
if not issubclass(getattr(grid, attr_name).__class__, FancyArray):
34-
setattr(grid, attr_name, attr_values)
34+
expected_type = grid.__dataclass_fields__[attr_name].type
35+
cast_value = expected_type(attr_values)
36+
setattr(grid, attr_name, cast_value)
3537
continue
3638

3739
try:
@@ -73,7 +75,7 @@ def _save_grid_to_json(
7375

7476
field_value = getattr(grid, field.name)
7577
if isinstance(field_value, (int, float, str, bool)):
76-
serialized_data[field.name] = field.type(field_value) # type: ignore
78+
serialized_data[field.name] = field_value
7779
continue
7880

7981
if not isinstance(field_value, FancyArray):

0 commit comments

Comments
 (0)