snake_case mem_per_cpu in SlurmIO (#44)

Matgenix · Aug 7, 2024 · 833f1f0 · 833f1f0
1 parent 346c0d6
commit 833f1f0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/qtoolkit/io/slurm.py b/src/qtoolkit/io/slurm.py
@@ -555,7 +555,7 @@ def _convert_time_to_str(time: int | float | timedelta) -> str:
     _qresources_mapping = {
         "queue_name": "partition",
         "job_name": "job_name",
-        "memory_per_thread": "mem-per-cpu",
+        "memory_per_thread": "mem_per_cpu",
         "account": "account",
         "qos": "qos",
         "priority": "priority",

diff --git a/tests/io/test_slurm.py b/tests/io/test_slurm.py
@@ -224,7 +224,7 @@ def test_check_convert_qresources(self, slurm_io):
         assert header_dict == {
             "partition": "myqueue",
             "job_name": "myjob",
-            "mem-per-cpu": 2048,
+            "mem_per_cpu": 2048,
             "account": "myaccount",
             "qos": "myqos",
             "qout_path": "someoutputpath",