amend

pytorch · Feb 27, 2024 · 48757d0 · 48757d0
1 parent e408e89
commit 48757d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_modules.py b/test/test_modules.py
@@ -1049,10 +1049,10 @@ def test_multiagent_cnn(
                     assert not torch.allclose(out[..., i, :], out[..., j, :])
 
     def test_multiagent_cnn_lazy(self):
-        n_channels = 5
+        n_agents = 5
         n_channels = 3
         cnn = MultiAgentConvNet(
-            n_agents=5,
+            n_agents=n_agents,
             centralised=False,
             share_params=False,
             in_features=None,