jupyter/spawner
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Enforce cluster flag in sinfo query

Browse source
This commit is contained in:
Benoît Seignovert 2024-03-22 19:41:27 +01:00
parent c0e95c4f55
commit aaf16ee89d
Signed by: Benoît Seignovert
GPG key ID: F5D8895227D18A0B
6 changed files with 26 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

2
render/__main__.py
View file

@ -32,7 +32,7 @@ SINFO = sinfo_from_file(
)
# Single vs. multi-cluster implementation
SLURM_SINGLE_CLUSTER = {'N/A': SINFO.pop('N/A')}
SLURM_SINGLE_CLUSTER = {'cluster': SINFO.pop('cluster')}
SLURM_MULTI_CLUSTER = SINFO
GPU_SINGLE_CLUSTER = gpu_max_duration(gres(SLURM_SINGLE_CLUSTER))
GPU_MULTI_CLUSTER = gpu_max_duration(gres(SLURM_MULTI_CLUSTER))