Skip to content
Closed
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 3 additions & 4 deletions src/math_spec_mapping/Load/type.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from .general import check_json_keys
from ..Classes import Type
import os
from typing import _UnionGenericAlias, List, _GenericAlias
from typing import List, get_origin
Comment thread
rororowyourboat marked this conversation as resolved.
Outdated


def convert_type(data, ms):
Expand Down Expand Up @@ -34,9 +34,8 @@ def convert_type(data, ms):
val = val.__name__
out[key] = val
data["type_name"]["python"] = str(out)
elif type(data["type"]["python"]) == _UnionGenericAlias:
data["type_name"]["python"] = data["type"]["python"].__repr__()
elif type(data["type"]["python"]) == _GenericAlias:
elif get_origin(data["type"]["python"]) is not None:
# Handle generic types (Union, List, etc.)
data["type_name"]["python"] = data["type"]["python"].__repr__()
else:
data["type_name"]["python"] = data["type"]["python"].__name__
Expand Down
Loading