Skip to content

Commit

Permalink
amend
Browse files Browse the repository at this point in the history
  • Loading branch information
matteobettini committed Apr 5, 2024
1 parent 627c8ff commit 4812cce
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion benchmarl/models/cnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ def __init__(
device=self.device,
**cnn_net_kwargs,
)
if hasattr(self.cnn, "_empy_net"):
if "_empty_net" in self.cnn.__dict__:
example_net = self.cnn._empty_net
else:
example_net = self.cnn.agent_networks[0]
Expand Down

0 comments on commit 4812cce

Please sign in to comment.