small fix

intsystems · Nov 21, 2024 · 3aa1ca8 · 3aa1ca8
1 parent 989f3eb
commit 3aa1ca8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/relaxit/distributions/LogisticNormalSoftmax.py b/src/relaxit/distributions/LogisticNormalSoftmax.py
@@ -30,7 +30,7 @@ def __init__(self, loc, scale, validate_args=None):
         )
 
     def expand(self, batch_shape, _instance=None):
-        new = self._get_checked_instance(LogisticNormal, _instance)
+        new = self._get_checked_instance(LogisticNormalSoftmax, _instance)
         return super().expand(batch_shape, _instance=new)
 
     @property