Do not remap storage in fgraph_to_python

aesara-devs · Nov 10, 2022 · 3415ab0 · 3415ab0
1 parent d10eafa
commit 3415ab0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aesara/link/utils.py b/aesara/link/utils.py
@@ -730,9 +730,9 @@ def fgraph_to_python(
 
     if order is None:
         order = fgraph.toposort()
-    input_storage, output_storage, storage_map = map_storage(
-        fgraph, order, input_storage, output_storage, storage_map
-    )
+
+    if input_storage is None or output_storage is None or storage_map is None:
+        input_storage, output_storage, storage_map = map_storage(fgraph, order)
 
     unique_name = unique_name_generator([fgraph_name])