Skip to content

Commit e7d39ed

Browse files
Fix Pandas downcast
1 parent 9229caf commit e7d39ed

1 file changed

Lines changed: 5 additions & 6 deletions

File tree

accelforge/mapper/FFM/_join_pmappings/pmapping_dataframe.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -631,16 +631,15 @@ def _adjust_reservations_one_resource(
631631
):
632632
target = nameloop2col(resource, level, left=left)
633633
if target in self.data:
634-
self.data.loc[:, target] += size
634+
add_to_col(self.data, target, size)
635635
continue
636636

637637
# We're creating a new column, so copy allocations from any parents
638638
source = self.get_reservation_or_parent(resource, level - 1)
639-
try:
640-
self.data[target] = size + (self.data[source] if source else 0)
641-
except:
642-
source = self.get_reservation_or_parent(resource, level - 1)
643-
self.data[target] = size + (self.data[source] if source else 0)
639+
if source:
640+
add_to_col(self.data, target, source)
641+
else:
642+
self.data[target] = size
644643

645644
# Assert all reservations are >= 0
646645
assert (self.data[target] >= 0).all(), f"Negative reservation: {target}"

0 commit comments

Comments
 (0)