From 4812ccec62a80fad3aac85e9b7a634e23cd8d460 Mon Sep 17 00:00:00 2001
From: Matteo Bettini <mb2389@cl.cam.ac.uk>
Date: Fri, 5 Apr 2024 19:12:46 +0200
Subject: [PATCH] amend

---
 benchmarl/models/cnn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmarl/models/cnn.py b/benchmarl/models/cnn.py
index 35aa8f57..b2a54cdb 100644
--- a/benchmarl/models/cnn.py
+++ b/benchmarl/models/cnn.py
@@ -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]