darabos commited on
Commit
56ec4e5
·
1 Parent(s): a3258ff

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