We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8797b09 commit 38a9b86Copy full SHA for 38a9b86
openrl/modules/networks/MAT_network.py
@@ -420,7 +420,7 @@ def get_actions(
420
):
421
obs = obs.reshape(-1, self.n_agent, self.obs_dim)
422
if action_masks is not None:
423
- action_masks = action_masks.reshape(-1, self.num_agents, self.action_dim)
+ action_masks = action_masks.reshape(-1, self.n_agent, self.action_dim)
424
425
# state unused
426
ori_shape = np.shape(obs)
0 commit comments