Slurm restapi v0.0.40 environment variables were not being found

DiamondLightSource · Aug 13, 2024 · b40b507 · b40b507
1 parent 83f1fb4
commit b40b507
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/cryoemservices/util/slurm_submission.py b/src/cryoemservices/util/slurm_submission.py
@@ -116,9 +116,9 @@ def slurm_submission(
         "current_working_directory": str(project_dir),
     }
     if api_version == "v0.0.38":
-        slurm_config["environment"] = {"USER": user, "HOME": user_home}
+        slurm_config["environment"] = {"USER": user, "HOME": user_home}  # type: ignore
     else:
-        slurm_config["environment"] = [f"USER: {user}", f"HOME: {user_home}"]
+        slurm_config["environment"] = [f"USER={user}", f"HOME={user_home}"]  # type: ignore
 
     # Add slurm partition and cluster preferences if given
     if slurm_rest.get("partition"):
@@ -138,7 +138,7 @@ def slurm_submission(
             slurm_json_job["memory_per_gpu"] = memory_request
         else:
             slurm_json_job["tres_per_task"] = "gres/gpu:1"
-            slurm_json_job["memory_per_node"]["number"] = memory_request
+            slurm_json_job["memory_per_node"]["number"] = memory_request  # type: ignore
     elif api_version == "v0.0.38":
         slurm_json_job["memory_per_cpu"] = 1000