Skip to content
Snippets Groups Projects

Use get_type_of_expression in typing_form_sympy_inspection to infer types

1 file
+ 4
2
Compare changes
  • Side-by-side
  • Inline
+ 4
2
@@ -436,8 +436,10 @@ def get_type_of_expression(expr,
@@ -436,8 +436,10 @@ def get_type_of_expression(expr,
elif isinstance(expr, TypedSymbol):
elif isinstance(expr, TypedSymbol):
return expr.dtype
return expr.dtype
elif isinstance(expr, sp.Symbol):
elif isinstance(expr, sp.Symbol):
return symbol_type_dict[expr.name]
if symbol_type_dict:
# raise ValueError("All symbols iside this expression have to be typed! ", str(expr))
return symbol_type_dict[expr.name]
 
else:
 
raise ValueError("All symbols inside this expression have to be typed! ", str(expr))
elif isinstance(expr, cast_func):
elif isinstance(expr, cast_func):
return expr.args[1]
return expr.args[1]
elif isinstance(expr, (vec_any, vec_all)):
elif isinstance(expr, (vec_any, vec_all)):
Loading