Update binners.py

coin-or · Apr 1, 2024 · 7c7aea7 · 7c7aea7
1 parent 68793ee
commit 7c7aea7
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/prtpy/binners.py b/prtpy/binners.py
@@ -75,7 +75,13 @@ def add_item_to_bin(self, bins:BinsArray, item: Any, bin_index: int)->BinsArray:
         Return the bins after the addition.
         """
         return bins
-
+
+    def remove_item_from_bin(self, bins:BinsArray, bin_index: int, item_index: int)->BinsArray:
+        sums, lists = bins
+        value = lists[bin_index][item_index]
+        sums[bin_index] -= value
+        del lists[bin_index][item_index]
+        return bins
 
     @abstractmethod
     def sort_by_ascending_sum(self, bins:BinsArray):
@@ -99,6 +105,8 @@ def numbins(self, bins: BinsArray) -> int:
         """
         return None
 
+
+
     @abstractmethod
     def sums(self, bins: BinsArray) -> Tuple[float]:
         """