diff options
Add missing import due to SymPy changes
| -rw-r--r-- | symbolic/optimizations.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/symbolic/optimizations.py b/symbolic/optimizations.py index 93dad09..8e43cc8 100644 --- a/symbolic/optimizations.py +++ b/symbolic/optimizations.py @@ -1,6 +1,7 @@ from sympy import * from sympy.codegen.rewriting import ReplaceOptim +from sympy.simplify import cse_main expand_square = ReplaceOptim( lambda e: e.is_Pow and e.exp.is_integer and e.exp == 2, |