fix: numpy and pandas classes did not consider axis parameter

xarray-contrib · Aug 25, 2022 · 8b75271 · 8b75271
1 parent 9e3210d
commit 8b75271
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/xeofs/models/eof.py b/xeofs/models/eof.py
@@ -97,7 +97,7 @@ def __init__(
  ):
 
  self._tf = _MultiArrayTransformer()
- X = self._tf.fit_transform(X)
+ X = self._tf.fit_transform(X, axis=axis)
  weights = self._tf.transform_weights(weights)
 
  super().__init__(

diff --git a/xeofs/models/mca.py b/xeofs/models/mca.py
@@ -92,8 +92,8 @@ def __init__(
 
  self._tfx = _MultiArrayTransformer()
  self._tfy = _MultiArrayTransformer()
- X = self._tfx.fit_transform(X)
- Y = self._tfy.fit_transform(Y)
+ X = self._tfx.fit_transform(X, axis=axis)
+ Y = self._tfy.fit_transform(Y, axis=axis)
  weights_X = self._tfx.transform_weights(weights_X)
  weights_Y = self._tfy.transform_weights(weights_Y)
 

diff --git a/xeofs/pandas/eof.py b/xeofs/pandas/eof.py
@@ -81,7 +81,7 @@ def __init__(
  ):
 
  self._tf = _MultiDataFrameTransformer()
- X = self._tf.fit_transform(X)
+ X = self._tf.fit_transform(X, axis=axis)
  weights = self._tf.transform_weights(weights)
 
  super().__init__(

diff --git a/xeofs/pandas/mca.py b/xeofs/pandas/mca.py
@@ -77,8 +77,8 @@ def __init__(
 
  self._tfx = _MultiDataFrameTransformer()
  self._tfy = _MultiDataFrameTransformer()
- X = self._tfx.fit_transform(X)
- Y = self._tfy.fit_transform(Y)
+ X = self._tfx.fit_transform(X, axis=axis)
+ Y = self._tfy.fit_transform(Y, axis=axis)
  weights_X = self._tfx.transform_weights(weights_X)
  weights_Y = self._tfy.transform_weights(weights_Y)