Coverage for src/bob/bio/base/config/dummy/pipeline.py: 92%

1from sklearn.base import BaseEstimator 

2 

3from bob.bio.base.algorithm import Distance 

4from bob.bio.base.pipelines import PipelineSimple 

5from bob.pipelines import wrap 

6 

7 

8class DummyTransformer(BaseEstimator): 

9 def fit(self, X, y=None): 

10 return self 

11 

12 def transform(self, X): 

13 return [X[0].mean(axis=1)] 

14 

15 

16transformer = wrap(["sample"], DummyTransformer()) 

17biometric_algorithm = Distance() 

18 

19pipeline = PipelineSimple(transformer, biometric_algorithm)