aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--symbolic/optimizations.py1
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,