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

Add slurm partition eq comparison

Browse source
This commit is contained in:
Benoît Seignovert 2024-02-19 17:40:57 +01:00
parent 4eef9d7016
commit e8ae32cc86
Signed by: Benoît Seignovert
GPG key ID: F5D8895227D18A0B
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/glicid_spawner/slurm.py
View file

@ -82,6 +82,9 @@ class SlurmPartition:
def __len__(self): def __len__(self):
return len(self.nodes) return len(self.nodes)
def __eq__(self, other):
return str(self) == str(other)
@property @property
def gpus(self) -> str: def gpus(self) -> str:
"""List of GPUs available.""" """List of GPUs available."""

3
tests/test_slurm.py
View file

@ -72,6 +72,7 @@ def test_slurm_dataclasses():
assert str(partition) == 'standard' # = name assert str(partition) == 'standard' # = name
assert partition.name == 'standard' assert partition.name == 'standard'
assert partition == 'standard' # __eq__ on name
assert len(partition) == 1 assert len(partition) == 1
@ -87,7 +88,7 @@ def test_slurm_dataclasses():
assert str(cluster) == 'nautilus' # = name assert str(cluster) == 'nautilus' # = name
assert cluster.name == 'nautilus' assert cluster.name == 'nautilus'
assert cluster == 'nautilus' # __eq__ assert cluster == 'nautilus' # __eq__ on name
assert len(cluster) == 1 assert len(cluster) == 1