Bugfix: Actually run disk-quota in mila code

Signed-off-by: Fabrice Normandin <normandf@mila.quebec>
mila-iqia · Sep 25, 2023 · 60f419e · 60f419e
1 parent 0812760
commit 60f419e
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/milatools/cli/commands.py b/milatools/cli/commands.py
@@ -542,14 +542,7 @@ def code(
     if command is None:
         command = os.environ.get("MILATOOLS_CODE_COMMAND", "code")
 
-        check_disk_quota(remote)
-
-        cnode = _find_allocation(
-            remote, job=job, node=node, alloc=alloc, job_name="mila-code"
-        )
-        if persist:
-            cnode = cnode.persist()
-        data, proc = cnode.ensure_allocation()
+    check_disk_quota(remote)
 
     cnode = _find_allocation(
         remote, job_name="mila-code", job=job, node=node, alloc=alloc