Skip to content

Commit 6984144

Browse files
authored
Fix PrioritizedSimpleReplay batch_update bug (#31)
1 parent fbe9fd3 commit 6984144

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

UtilsRL/rl/buffer/prioritized_replay.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def batch_update(self, batch_idx, metric_value):
9393
if metric_value.shape == ():
9494
metric_value = np.asarray([metric_value, ])
9595
# update crendential
96-
self.max_metric_value = max(metric_value, self.max_metric_value)
96+
self.max_metric_value = max(np.max(metric_value), self.max_metric_value)
9797
self.sum_tree.update(batch_idx, self.metric_fn(metric_value))
9898
self.min_tree.update(batch_idx, self.metric_fn(metric_value))
9999

0 commit comments

Comments
 (0)