Skip to content

Commit ee73ff3

Browse files
committed
Reservation: sync tres and cpus/node_count attributes
1 parent ff205f7 commit ee73ff3

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

pyslurm/core/reservation.pyx

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,9 @@ cdef class Reservation:
340340
@cpus.setter
341341
def cpus(self, val):
342342
self.info.core_cnt = self.umsg.core_cnt = int(val)
343+
current_tres = self.tres
344+
current_tres["cpu"] = int(val)
345+
self.tres = current_tres
343346

344347
@property
345348
def cpu_ids_by_node(self):
@@ -410,6 +413,9 @@ cdef class Reservation:
410413
@node_count.setter
411414
def node_count(self, val):
412415
self.info.node_cnt = self.umsg.node_cnt = int(val)
416+
current_tres = self.tres
417+
current_tres["node"] = int(val)
418+
self.tres = current_tres
413419

414420
@property
415421
def nodes(self):
@@ -477,6 +483,13 @@ cdef class Reservation:
477483
def tres(self, val):
478484
cstr.fmalloc2(&self.info.tres_str, &self.umsg.tres_str,
479485
cstr.dict_to_str(val))
486+
current = self.tres
487+
cpus, node_count = self.tres.get("cpu"), self.tres.get("node")
488+
if cpus and self.cpus != cpus:
489+
self.cpus = cpus
490+
491+
if node_count and self.node_count != node_count:
492+
self.node_count = node_count
480493

481494
@property
482495
def users(self):

0 commit comments

Comments
 (0)