1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00

Wolpertinger preset failure fix (#434)

Numpy 1.18 fails to cast float to int as part of the wolpertinger preset run
This commit is contained in:
Gal Novik
2020-01-14 16:26:38 +02:00
committed by GitHub
parent 525a22cb5b
commit 79b05a8105

View File

@@ -59,7 +59,7 @@ class BoxDiscretization(PartialDiscreteActionSpaceMap):
def get_unfiltered_action_space(self, output_action_space: BoxActionSpace) -> DiscreteActionSpace:
if isinstance(self.num_bins_per_dimension, int):
self.num_bins_per_dimension = np.ones(output_action_space.shape) * self.num_bins_per_dimension
self.num_bins_per_dimension = np.ones(output_action_space.shape).astype(int) * self.num_bins_per_dimension
bins = []
for i in range(len(output_action_space.low)):