fix #23 maybe

amakelov · Aug 24, 2024 · b9c693e · b9c693e
1 parent 25915ab
commit b9c693e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mandala/storage.py b/mandala/storage.py
@@ -769,7 +769,8 @@ def call_internal(
             op=op,
             pre_call_uid=pre_call_id,
             inputs=wrapped_inputs,
-            code_state=self.guess_code_state() if must_version_call else None,
+            # code_state=self.guess_code_state() if must_version_call else None,
+            code_state = self.code_state if must_version_call else None,
             versioner=self.cached_versioner if must_version_call else None,
             must_version=must_version_call,
         )