Cast MultiCategorical num_outputs to int. (#4629)

2026-07-02 10:12:04 +08:00 · 2019-04-15 03:51:37 +01:00
parent 967e8aad9d
commit f600591468
1 changed files with 1 additions and 1 deletions
@@ -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))