From 8c207f1424f8865d7e9af990326000b3dc733db4 Mon Sep 17 00:00:00 2001 From: Adrian Kummerlaender Date: Wed, 16 Jun 2021 23:03:52 +0200 Subject: Add missing import due to SymPy changes --- symbolic/optimizations.py | 1 + 1 file changed, 1 insertion(+) (limited to 'symbolic') 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, -- cgit v1.2.3