Skip to content

Commit 6bde766

Browse files
committed
fix(load): handle GPU without model
1 parent 3f2f3d0 commit 6bde766

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

firehpc/load.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def _get_partition_gpus(self, partition):
158158
gres = gres.split(":")
159159
if gres[0] != "gpu":
160160
continue
161-
result += int(gres[2])
161+
result += int(gres[-1])
162162
except json.decoder.JSONDecodeError as err:
163163
raise FireHPCRuntimeError(
164164
f"Unable to retrieve nodes from cluster {self.cluster.name}: {str(err)}"

0 commit comments

Comments
 (0)