From 1420edcfda189582497e327d7c57f7aa36c9dbba Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Thu, 25 Jul 2024 15:53:32 +0100 Subject: [PATCH] fix --- torchrl/modules/models/multiagent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchrl/modules/models/multiagent.py b/torchrl/modules/models/multiagent.py index 126f230e980..6ccc4721678 100644 --- 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 * " ")