aboutsummaryrefslogtreecommitdiff
path: root/symbolic
diff options
context:
space:
mode:
authorAdrian Kummerlaender2021-06-16 23:03:52 +0200
committerAdrian Kummerlaender2021-06-16 23:03:52 +0200
commit8c207f1424f8865d7e9af990326000b3dc733db4 (patch)
tree7a7a4eb5805995001419f8b7774a3cbcf2fe72d0 /symbolic
parent4c8fdec9be6d42d8d94b43bdcf1b0d7ada030a46 (diff)
downloadsymlbm_playground-8c207f1424f8865d7e9af990326000b3dc733db4.tar
symlbm_playground-8c207f1424f8865d7e9af990326000b3dc733db4.tar.gz
symlbm_playground-8c207f1424f8865d7e9af990326000b3dc733db4.tar.bz2
symlbm_playground-8c207f1424f8865d7e9af990326000b3dc733db4.tar.lz
symlbm_playground-8c207f1424f8865d7e9af990326000b3dc733db4.tar.xz
symlbm_playground-8c207f1424f8865d7e9af990326000b3dc733db4.tar.zst
symlbm_playground-8c207f1424f8865d7e9af990326000b3dc733db4.zip
Add missing import due to SymPy changes
Diffstat (limited to 'symbolic')
-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,