Coverage for src/bob/pipelines/__init__.py: 100%

1# isort: skip_file 

2from . import distributed # noqa: F401 

3from . import transformers # noqa: F401 

4from . import utils # noqa: F401 

5from . import xarray as xr # noqa: F401 

6from .sample import ( 

7 DelayedSample, 

8 DelayedSampleSet, 

9 DelayedSampleSetCached, 

10 Sample, 

11 SampleBatch, 

12 SampleSet, 

13) 

14from .utils import ( # noqa: F401 

15 assert_picklable, 

16 check_parameter_for_validity, 

17 check_parameters_for_validity, 

18 flatten_samplesets, 

19 hash_string, 

20 is_picklable, 

21) 

22from .wrappers import wrap # noqa: F401 

23from .wrappers import ( # noqa: F401 

24 BaseWrapper, 

25 CheckpointWrapper, 

26 DaskWrapper, 

27 DelayedSamplesCall, 

28 SampleWrapper, 

29 ToDaskBag, 

30 dask_tags, 

31 estimator_requires_fit, 

32 get_bob_tags, 

33 getattr_nested, 

34 is_instance_nested, 

35 is_pipeline_wrapped, 

36) 

37from .dataset import FileListToSamples, CSVToSamples, FileListDatabase 

38 

39 

40def __appropriate__(*args): 

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

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

43 shortened. 

44 

45 Parameters 

46 ---------- 

47 *args 

48 The objects that you want sphinx to believe that are defined here. 

49 

50 Resolves `Sphinx referencing issues <https//github.com/sphinx- 

51 doc/sphinx/issues/3048>` 

52 """ 

53 

54 for obj in args: 

55 obj.__module__ = __name__ 

56 

57 

58__appropriate__( 

59 Sample, 

60 DelayedSample, 

61 SampleSet, 

62 DelayedSampleSet, 

63 DelayedSampleSetCached, 

64 SampleBatch, 

65 BaseWrapper, 

66 DelayedSamplesCall, 

67 SampleWrapper, 

68 CheckpointWrapper, 

69 DaskWrapper, 

70 ToDaskBag, 

71 FileListToSamples, 

72 CSVToSamples, 

73 FileListDatabase, 

74) 

75 

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

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