fix: fix dimensional analysis regression

Check wasn't working for non-sympy dimension exponents
mgreminger · Jan 11, 2025 · bec2e6e · bec2e6e
1 parent 1e23b20
commit bec2e6e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/public/dimensional_analysis.py b/public/dimensional_analysis.py
@@ -780,6 +780,11 @@ def custom_get_dimensional_dependencies_for_name(self, dimension):
             for d in dicts:
                 keys_to_remove = set()
                 for key, exp in d.items():
+                    if isinstance(exp, int):
+                        exp = sympify(float(exp))
+                    elif isinstance(exp, float):
+                        exp = sympify(exp)
+
                     new_exp = exp.round(EXP_NUM_DIGITS) 
                     if new_exp == sympify("0"):
                         keys_to_remove.add(key)