1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +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: def get_unfiltered_action_space(self, output_action_space: BoxActionSpace) -> DiscreteActionSpace:
if isinstance(self.num_bins_per_dimension, int): 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 = [] bins = []
for i in range(len(output_action_space.low)): for i in range(len(output_action_space.low)):