diff --git a/src/qtoolkit/io/slurm.py b/src/qtoolkit/io/slurm.py index e555092..e415e8c 100644 --- 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 index f3ffe00..1b92dae 100644 --- 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",