[rllib] Fix error in shape calculation. (#7301)

2026-08-11 11:24:51 +08:00 · 2020-02-25 14:16:29 -08:00
parent f14b6e477b
commit 75f683eec6
1 changed files with 1 additions and 0 deletions
@@ -229,6 +229,7 @@ class ModelCatalog:
                else:
                    all_discrete = False
                    size += np.product(action_space.spaces[i].shape)
+            size = int(size)
            return (tf.int64 if all_discrete else tf.float32, (None, size))
        elif isinstance(action_space, gym.spaces.Dict):
            raise NotImplementedError(