From 491abece229d0b67ced7d28c2289de27a7a25aa0 Mon Sep 17 00:00:00 2001 From: ottofabian Date: Mon, 8 Aug 2022 08:21:28 +0200 Subject: [PATCH] Update black_box_wrapper.py --- fancy_gym/black_box/black_box_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fancy_gym/black_box/black_box_wrapper.py b/fancy_gym/black_box/black_box_wrapper.py index df2206a..119d923 100644 --- a/fancy_gym/black_box/black_box_wrapper.py +++ b/fancy_gym/black_box/black_box_wrapper.py @@ -87,7 +87,7 @@ class BlackBoxWrapper(gym.ObservationWrapper): def _get_traj_gen_action_space(self): """This function can be used to set up an individual space for the parameters of the traj_gen.""" - min_action_bounds, max_action_bounds = self.traj_gen.get_params_bounds().t() + min_action_bounds, max_action_bounds = self.traj_gen.get_params_bounds() action_space = gym.spaces.Box(low=min_action_bounds.numpy(), high=max_action_bounds.numpy(), dtype=self.env.action_space.dtype) return action_space