fix(cross): fix data augmentation in Hilbert cross-set models (#222)

xarray-contrib · Sep 10, 2024 · 960f744 · 960f744
1 parent 75bbe7a
commit 960f744
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xeofs/cross/base_model_cross_set.py b/xeofs/cross/base_model_cross_set.py
@@ -298,7 +298,7 @@ def fit(
         X = self.whitener1.fit_transform(X)
         Y = self.whitener2.fit_transform(Y)
         # Augment data
-        X, y = self._augment_data(X, Y)
+        X, Y = self._augment_data(X, Y)
         # Fit the model
         self._fit_algorithm(X, Y)