Coverage for src/bob/bio/base/transformers/__init__.py: 67%

15 statements  

« prev     ^ index     » next       coverage.py v7.6.0, created at 2024-07-12 22:34 +0200

1# isort: skip_file 

2from collections import defaultdict 

3 

4 

5def split_X_by_y(X, y): 

6 training_data = defaultdict(list) 

7 for x1, y1 in zip(X, y): 

8 training_data[y1].append(x1) 

9 training_data = list(training_data.values()) 

10 return training_data 

11 

12 

13from .preprocessor import PreprocessorTransformer 

14from .extractor import ExtractorTransformer 

15from .preprocessing import ReferenceIdEncoder 

16 

17 

18# gets sphinx autodoc done right - don't remove it 

19def __appropriate__(*args): 

20 """Says object was actually declared here, and not in the import module. 

21 Fixing sphinx warnings of not being able to find classes, when path is 

22 shortened. 

23 

24 Parameters 

25 ---------- 

26 *args 

27 An iterable of objects to modify 

28 

29 Resolves `Sphinx referencing issues 

30 <https://github.com/sphinx-doc/sphinx/issues/3048>` 

31 """ 

32 

33 for obj in args: 

34 obj.__module__ = __name__ 

35 

36 

37__appropriate__( 

38 PreprocessorTransformer, 

39 ExtractorTransformer, 

40 ReferenceIdEncoder, 

41) 

42 

43__all__ = [_ for _ in dir() if not _.startswith("_")]