Re-add check for "/" prefix

(not sure why or how it got removed in the first place tbh) Signed-off-by: Fabrice Normandin <normandf@mila.quebec>
mila-iqia · Sep 19, 2023 · 8fdfeac · 8fdfeac
1 parent 1b6a509
commit 8fdfeac
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/milatools/cli/commands.py b/milatools/cli/commands.py
@@ -304,6 +304,11 @@ def code():
 
         node_name = data["node_name"]
 
+        if not path.startswith("/"):
+            # Get $HOME because we have to give the full path to code
+            home = remote.home()
+            path = "/".join([home, path])
+
         try:
             while True:
                 here.run(
@@ -408,8 +413,7 @@ def list():
                     necessary_keys = {"node_name", "to_forward"}
                     if any(k not in info for k in necessary_keys):
                         qn.print(
-                            f"{identifier} ({program}, MISSING INFO)",
-                            style="bold red",
+                            f"{identifier} ({program}, MISSING INFO)", style="bold red"
                         )
                         to_purge.append((identifier, jobid))
                     else: