From dc774e963179b23e49e136c7a8a12882578020b4 Mon Sep 17 00:00:00 2001 From: Salem Date: Thu, 31 Aug 2023 21:22:53 -0400 Subject: [PATCH] fix __repr__ of modules - no more env --- src/gfn/modules.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gfn/modules.py b/src/gfn/modules.py index 1dde4fa4..13080c7f 100644 --- 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