Spaces:
Running
Running
Better error message for wrong enum value.
Browse files
lynxkite-core/src/lynxkite/core/ops.py
CHANGED
@@ -144,6 +144,7 @@ def _param_to_type(name, value, type):
|
|
144 |
assert value != "", f"{name} is unset."
|
145 |
return float(value)
|
146 |
if isinstance(type, enum.EnumMeta):
|
|
|
147 |
return type[value]
|
148 |
if isinstance(type, types.UnionType):
|
149 |
match type.__args__:
|
|
|
144 |
assert value != "", f"{name} is unset."
|
145 |
return float(value)
|
146 |
if isinstance(type, enum.EnumMeta):
|
147 |
+
assert value in type.__members__, f"{value} is not an option for {name}."
|
148 |
return type[value]
|
149 |
if isinstance(type, types.UnionType):
|
150 |
match type.__args__:
|