handle export_memory_steps being None (#63)

EmbarkStudios · Jun 2, 2023 · 23ac61b · 23ac61b
1 parent ef34798
commit 23ac61b
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/py/erupt-ark/erupt_ark/legacy.py b/src/py/erupt-ark/erupt_ark/legacy.py
@@ -227,12 +227,14 @@ def construct(
                 ),
                 1000,
             )
-            memory_proxy = MemoryExporterProxyWrapper(
-                memory=memory_proxy,
-                target_memory_name=data_loader.data_group,
-                inf_steps_per_memory_export=conf.export_memory_steps,
-                experiment_root_path=engine_config.artifact_location,
-            )
+
+            if conf.export_memory_steps is not None:
+                memory_proxy = MemoryExporterProxyWrapper(
+                    memory=memory_proxy,
+                    target_memory_name=data_loader.data_group,
+                    inf_steps_per_memory_export=conf.export_memory_steps,
+                    experiment_root_path=engine_config.artifact_location,
+                )
 
         num_actions = spaces.actions.shape[0]
         assert (