Skip to content

Commit 2deac45

Browse files
committed
Merge branch 'main' of github.com:OpenRL-Lab/openrl
2 parents b6bcb70 + 38a9b86 commit 2deac45

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

openrl/modules/networks/MAT_network.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -420,7 +420,7 @@ def get_actions(
420420
):
421421
obs = obs.reshape(-1, self.n_agent, self.obs_dim)
422422
if action_masks is not None:
423-
action_masks = action_masks.reshape(-1, self.num_agents, self.action_dim)
423+
action_masks = action_masks.reshape(-1, self.n_agent, self.action_dim)
424424

425425
# state unused
426426
ori_shape = np.shape(obs)

0 commit comments

Comments
 (0)