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

Add slurm cluster and partition len property

Browse source
This commit is contained in:
Benoît Seignovert 2024-02-19 17:32:23 +01:00
parent fc54701400
commit 4eef9d7016
Signed by: Benoît Seignovert
GPG key ID: F5D8895227D18A0B
2 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/glicid_spawner/slurm.py
View file

@ -79,6 +79,9 @@ class SlurmPartition:
def __iter__(self):
return iter(self.nodes)
def __len__(self):
return len(self.nodes)
@property
def gpus(self) -> str:
"""List of GPUs available."""
@ -108,6 +111,9 @@ class SlurmCluster:
def __iter__(self):
return iter(self.partitions)
def __len__(self):
return len(self.partitions)
def __eq__(self, other):
return str(self) == str(other)