fix __repr__ of modules - no more env

GFNOrg · Sep 1, 2023 · dc774e9 · dc774e9
1 parent b59a507
commit dc774e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gfn/modules.py b/src/gfn/modules.py
@@ -81,7 +81,7 @@ def forward(self, states: States) -> TT["batch_shape", "output_dim", float]:
         return out
 
     def __repr__(self):
-        return f"{self.__class__.__name__}({self.env})"
+        return f"{self.__class__.__name__} module"
 
     @property
     @abstractmethod