fix

pytorch · Jul 25, 2024 · 1420edc · 1420edc
1 parent 6268443
commit 1420edc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/modules/models/multiagent.py b/torchrl/modules/models/multiagent.py
@@ -240,7 +240,7 @@ def from_stateful_net(self, stateful_net: nn.Module):
         self.params.data.update_(params.data)
 
     def __repr__(self):
-        empty_net = self.__dict__["_empty_net"]
+        empty_net = self._empty_net
         with self.params.to_module(empty_net):
             module_repr = indent(str(empty_net), 4 * " ")
         n_agents = indent(f"n_agents={self.n_agents}", 4 * " ")