Add slurm partition eq comparison

2024-02-19 17:40:57 +01:00 · 2024-02-19 17:40:57 +01:00 · e8ae32cc86
commit e8ae32cc86
parent 4eef9d7016
2 changed files with 5 additions and 1 deletions
--- a/src/glicid_spawner/slurm.py
+++ b/src/glicid_spawner/slurm.py
@ -82,6 +82,9 @@ class SlurmPartition:
    def __len__(self):
        return len(self.nodes)

+    def __eq__(self, other):
+        return str(self) == str(other)
+
    @property
    def gpus(self) -> str:
        """List of GPUs available."""