From b9c15a25513eb13df87484bb42c790e0300836d7 Mon Sep 17 00:00:00 2001 From: Stanislav Chekmenev Date: Tue, 2 Feb 2021 13:13:43 +0100 Subject: [PATCH] [RLlib] Issue #13761: Fix get action shape (#13764) --- rllib/policy/policy.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rllib/policy/policy.py b/rllib/policy/policy.py index 577ac3d68..1bce4b96d 100644 --- a/rllib/policy/policy.py +++ b/rllib/policy/policy.py @@ -709,7 +709,8 @@ class Policy(metaclass=ABCMeta): ret = {} for view_col, view_req in self.view_requirements.items(): if isinstance(view_req.space, (gym.spaces.Dict, gym.spaces.Tuple)): - _, shape = ModelCatalog.get_action_shape(view_req.space) + _, shape = ModelCatalog.get_action_shape( + view_req.space, framework=self.config["framework"]) ret[view_col] = \ np.zeros((batch_size, ) + shape[1:], np.float32) else: