check for cache kwarg in get_contraction() calls

jcmgray · Feb 21, 2022 · 36dedb7 · 36dedb7
1 parent 81f553d
commit 36dedb7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/quimb/tensor/tensor_core.py b/quimb/tensor/tensor_core.py
@@ -237,6 +237,10 @@ def get_contraction(eq, *shapes, cache=True, get='expr',
         expr = expr_fn(eq, *shapes, optimize=optimize, **kwargs)
         return expr
 
+    # check for user-supplied cache kwarg
+    if 'cache' in kwargs:
+        cache = kwargs['cache']
+
     # make sure shapes are hashable + concrete python ints
     if not (
         isinstance(shapes[0], tuple) and