small fixes

Change-Id: Ic44a24779e836f1fd96862b5b7a9cd8604e5e59f
BlueBrain · May 30, 2024 · c780aa1 · c780aa1
1 parent 539419c
commit c780aa1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bluepyemodel/tools/utils.py b/bluepyemodel/tools/utils.py
@@ -42,7 +42,7 @@ def checkpoint_path_exists(checkpoint_path):
 
 
 def get_checkpoint_path(metadata, seed=None):
-    """Get checkpoint path. Use legacy format if any is found, if not use latest format."""
+    """Get checkpoint path. Use legacy format if any is found, else use latest format."""
     base_path = f"./checkpoints/{metadata.emodel}/{metadata.iteration}/"
     # legacy case 1 (2023.05.11 - 2023.10.19)
     filename = metadata.as_string(
@@ -110,15 +110,15 @@ def yesno(question):
 
 
 def get_seed_from_checkpoint_path(path):
-    """"""
+    """Get seed from checkpoint path name. default seed is 0 if not found."""
 
     if path.endswith(".tmp"):
         path = path.replace(".tmp", "")
 
     filename = Path(path).stem.split("__")
 
     search_str = "seed="
-    seed = next((e.replace(search_str, "") for e in filename if search_str in e), None)
+    seed = next((e.replace(search_str, "") for e in filename if search_str in e), 0)
 
     return int(seed)
 

diff --git a/tests/unit_tests/test_optimisation.py b/tests/unit_tests/test_optimisation.py
@@ -24,7 +24,7 @@
 
 def checkpoint_check(dir, fname, metadata, inner_dir):
     f = dir / fname
-    f.write_text("")
+    f.touch()
     assert str(get_checkpoint_path(metadata, seed=0)) == "/".join((".", inner_dir, fname))
     f.unlink()