Fix class structure in infonce.py

berenslab · Dec 13, 2024 · eef7482 · eef7482
1 parent 2948391
commit eef7482
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tsimcne/losses/infonce.py b/tsimcne/losses/infonce.py
@@ -136,7 +136,7 @@ def compute(self):
         self.criterion = self.cls(**self.kwargs)
 
 
-class InfoNCET(InfoNCEZ):
+class InfoNCET(InfoNCEGaussian):
     def __init__(self, dof=None, **kwargs):
         super().__init__(**kwargs)
         self.dof = dof