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__:
|