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

Cast MultiCategorical num_outputs to int. (#4629)

Browse source
This commit is contained in:
Vlad Firoiu 2019-04-15 03:51:37 +01:00 • committed by Eric Liang
parent 967e8aad9d
commit f600591468
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/ray/rllib/models/catalog.py
View file

@ -147,7 +147,7 @@ class ModelCatalog(object):
elif isinstance(action_space, gym.spaces.multi_discrete.MultiDiscrete):
if torch:
raise NotImplementedError
return MultiCategorical, sum(action_space.nvec)
return MultiCategorical, int(sum(action_space.nvec))
raise NotImplementedError("Unsupported args: {} {}".format(
action_space, dist_type))