Coverage for src/io_collection/load/__init__.py: 89%

1"""Tasks for loading different object types from local file systems or S3 buckets.""" 

2 

3from prefect import task 

4 

5from .load_buffer import load_buffer 

6from .load_dataframe import load_dataframe 

7from .load_json import load_json 

8from .load_pickle import load_pickle 

9from .load_tar import load_tar 

10from .load_text import load_text 

11 

12load_buffer = task(load_buffer) 

13load_dataframe = task(load_dataframe) 

14load_json = task(load_json) 

15load_pickle = task(load_pickle) 

16load_tar = task(load_tar) 

17load_text = task(load_text) 

18 

19try: 

20 from .load_image import load_image 

21 

22 load_image = task(load_image) 

23except ImportError: 

24 pass