chore(release): 3.0.1 (#223)

xarray-contrib · Sep 10, 2024 · c2e15b6 · c2e15b6
2 parents 75bbe7a + 2dd3e86
commit c2e15b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
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)
 

diff --git a/xeofs/cross/cpcca.py b/xeofs/cross/cpcca.py
@@ -982,8 +982,8 @@ def _compute_total_squared_covariance(self, C: DataArray) -> DataArray:
 
         Requires the unwhitened covariance matrix which we can obtain by multiplying the whitened covariance matrix with the inverse of the whitening transformation matrix.
         """
-        C = self.whitener2.inverse_transform_data(C)
-        C = self.whitener1.inverse_transform_data(C.conj().T)
+        C = self.whitener2.inverse_transform_data(C, unwhiten_only=True)
+        C = self.whitener1.inverse_transform_data(C.conj().T, unwhiten_only=True)
         # Not necessary to conjugate transpose for total squared covariance
         # C = C.conj().T
         return (abs(C) ** 2).sum()