Merge pull request #135 from saleml/fixrepr

fix __repr__ of modules
GFNOrg · Sep 4, 2023 · ec4b415 · ec4b415
2 parents b59a507 + dc774e9
commit ec4b415
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