From eef7482457d249e052359cff32ee4c80d4db53be Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Niklas=20B=C3=B6hm?= <mail@jnboehm.com>
Date: Fri, 13 Dec 2024 13:36:30 +0100
Subject: [PATCH] Fix class structure in infonce.py

---
 tsimcne/losses/infonce.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tsimcne/losses/infonce.py b/tsimcne/losses/infonce.py
index e113a6e..61428db 100644
--- 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