diff --git a/src/glicid_spawner/spawner.py b/src/glicid_spawner/spawner.py index eb3cfdf..8906780 100644 --- a/src/glicid_spawner/spawner.py +++ b/src/glicid_spawner/spawner.py @@ -43,10 +43,10 @@ class GlicidSpawner(SlurmSpawner): # Export options options['cpus-per-task'] = CPU[i_cpu].description - options['mem'] = RAM[i_ram] + 'GB' + options['mem'] = RAM[i_ram].description + 'GB' if i_gpu: - options['gres'] = 'gpu:' + GPU[i_gpu] + options['gres'] = 'gpu:' + GPU[i_gpu].description options['time'] = f'{duration:02d}:00:00'