Merge branch 'Founding-Zero-main' into multipe_inputs_to_models

facebookresearch · Apr 5, 2024 · b412572 · b412572
2 parents d56a9c0 + 8f90b62
commit b412572
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/.github/unittest/install_dependencies_nightly.sh b/.github/unittest/install_dependencies_nightly.sh
@@ -11,7 +11,8 @@ python -m pip install torch
 
 cd ../BenchMARL
 pip install -e .
-pip uninstal torchrl tensordict
+pip uninstall --yes torchrl
+pip uninstall --yes tensordict
 
 cd ..
 python -m pip install git+https://github.com/pytorch-labs/tensordict.git

diff --git a/benchmarl/models/cnn.py b/benchmarl/models/cnn.py
@@ -137,10 +137,8 @@ def __init__(
                 device=self.device,
                 **cnn_net_kwargs,
             )
-            if "_empty_net" in self.cnn.__dict__:
-                example_net = self.cnn._empty_net
-            else:
-                example_net = self.cnn.agent_networks[0]
+            example_net = self.cnn._empty_net
+
         else:
             self.cnn = nn.ModuleList(
                 [