11import numpy as np
22from sklearn .base import BaseEstimator , TransformerMixin
3- from sklearn .utils .validation import check_is_fitted , check_array
3+ from sklearn .utils .validation import check_is_fitted , validate_data
44
55
6- class Delayer (BaseEstimator , TransformerMixin ):
6+ class Delayer (TransformerMixin , BaseEstimator ):
77 """Scikit-learn Transformer to add delays to features.
88
99 This assumes that the samples are ordered in time.
@@ -50,7 +50,7 @@ def fit(self, X, y=None):
5050 -------
5151 self : returns an instance of self.
5252 """
53- X = self . _validate_data ( X , dtype = 'numeric' )
53+ X = validate_data ( self , X , dtype = 'numeric' )
5454 self .n_features_in_ = X .shape [1 ]
5555 return self
5656
@@ -68,12 +68,9 @@ def transform(self, X):
6868 Transformed data.
6969 """
7070 check_is_fitted (self )
71- X = check_array ( X , copy = True )
71+ X = validate_data ( self , X , reset = False , copy = True )
7272
7373 n_samples , n_features = X .shape
74- if n_features != self .n_features_in_ :
75- raise ValueError (
76- 'Different number of features in X than during fit.' )
7774
7875 if self .delays is None :
7976 return X
0 commit comments