[RLlib] Issue #13761: Fix get action shape (#13764)

2026-07-22 13:00:49 +08:00 · 2021-02-02 13:13:43 +01:00
parent 714c367b9d
commit b9c15a2551
1 changed files with 2 additions and 1 deletions
@@ -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: