fix bug with dirty token and job arrays

facebookresearch · May 5, 2022 · 33bf261 · 33bf261
1 parent a23ce11
commit 33bf261
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dora/__init__.py b/dora/__init__.py
@@ -60,7 +60,7 @@
 __pdoc__ = {}
 __pdoc__['tests'] = False
 
-__version__ = "0.1.10a7"
+__version__ = "0.1.10a8"
 
 # flake8: noqa
 from .explore import Explorer, Launcher

diff --git a/dora/shep.py b/dora/shep.py
@@ -305,7 +305,7 @@ def _check_orphans(self):
                            "but Dora or Slurm crashed before the job id was saved.")
             proc = sp.run(["squeue", "-u", os.getlogin(), "-n", name, "-o", "%i", "-h"],
                           capture_output=True, check=True)
-            ids = [line for line in proc.stdout.decode().strip().split("\n") if line]
+            ids = [line.split('_')[0] for line in proc.stdout.decode().strip().split("\n") if line]
             if ids:
                 logger.warning(f"Found orphan job ids {ids}, will cancel")
                 sp.run(["scancel"] + ids, check=True)