Cast MultiCategorical num_outputs to int. (#4629)

This commit is contained in:
Vlad Firoiu authored and Eric Liang committed 2019-04-14 19:51:37 -07:00
1 parent 967e8aad9d
commit f600591468
1 file changed
+1 -1
+1 -1
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))