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:
@@ -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)):
|
||||||
|
|||||||
Reference in New Issue
Block a user