Coverage for src/io_collection/save/__init__.py: 91%

1"""Tasks for saving different object types to local file systems or S3 buckets.""" 

2 

3from prefect import task 

4 

5from .save_buffer import save_buffer 

6from .save_dataframe import save_dataframe 

7from .save_figure import save_figure 

8from .save_gif import save_gif 

9from .save_json import save_json 

10from .save_pickle import save_pickle 

11from .save_tar import save_tar 

12from .save_text import save_text 

13 

14save_buffer = task(save_buffer) 

15save_dataframe = task(save_dataframe) 

16save_figure = task(save_figure) 

17save_gif = task(save_gif) 

18save_json = task(save_json) 

19save_pickle = task(save_pickle) 

20save_tar = task(save_tar) 

21save_text = task(save_text) 

22 

23try: 

24 from .save_image import save_image 

25 

26 save_image = task(save_image) 

27except ImportError: 

28 pass