hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Stanislav Chekmenev 2021-02-02 13:13:43 +01:00 • committed by GitHub
parent 714c367b9d
commit b9c15a2551
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
rllib/policy/policy.py
View file

@ -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: