Skip to content

Commit 0c335b2

Browse files
committed
revert back to max item removals
1 parent b3ef397 commit 0c335b2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

weight_predict/weight_predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def predict_most_likely_item(self, weight_delta: float) -> List[Item]:
4848
# Store probabilities
4949
probabilities[item.item_id] = []
5050
# Iterate through all possible quantities
51-
for potential_quantity in range(1, item.quantity + 1):
51+
for potential_quantity in range(1, MAX_ITEM_REMOVALS_TO_CHECK + 1):
5252
expected_weight = item.avg_weight * potential_quantity
5353
scaled_std = item.std_weight * (potential_quantity ** 0.5)
5454

0 commit comments

Comments
 (0)