fix error in diff namer for overlapping output checkpointed experiments

octopuscabbage · Mar 12, 2021 · 693a962 · 693a962
1 parent 53d0042
commit 693a962
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@ def read(fname):
 
 setup(
     name="smallab",
-    version="1.11.4",
+    version="1.11.5",
     url='https://github.com/octopuscabbage/smallab',
     packages=find_packages(),
     install_requires=required,

diff --git a/smallab/runner/runner_methods.py b/smallab/runner/runner_methods.py
@@ -81,7 +81,8 @@ def _interior_fn():
         result = run_with_correct_handler(experiment, name, specification,eventQueue, diff_namer=diff_namer)
         if isinstance(result, types.GeneratorType):
             for cur_result in result:
-                diff_namer.extend_name(cur_result["specification"])
+                if diff_namer is not None:
+                    diff_namer.extend_name(cur_result["specification"])
                 save_run(name, experiment, cur_result["specification"], cur_result["result"], force_pickle,diff_namer=diff_namer, extended_keys=True)
         else:
             save_run(name, experiment, specification, result, force_pickle,diff_namer=diff_namer)