Redefined inverse method of Inverse wrapper class

luigibonati · Apr 4, 2024 · 551ad2e · 551ad2e
1 parent d7d8c8b
commit 551ad2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlcolvar/core/transform/utils.py b/mlcolvar/core/transform/utils.py
@@ -99,10 +99,10 @@ def __init__(self, module: torch.nn.Module):
         self.module = module
 
     def inverse(self, *args, **kwargs):
-        return self.module.inverse(*args, **kwargs)
+        return self.module(*args, **kwargs)
 
     def forward(self, *args, **kwargs):
-        return self.inverse(*args, **kwargs)
+        return self.module.inverse(*args, **kwargs)