diff --git a/clarifai_datautils/multimodal/pipeline/base.py b/clarifai_datautils/multimodal/pipeline/base.py index 5a0d7dd..6121ffa 100644 --- a/clarifai_datautils/multimodal/pipeline/base.py +++ b/clarifai_datautils/multimodal/pipeline/base.py @@ -3,8 +3,6 @@ from tqdm import tqdm -from clarifai_datautils.multimodal.pipeline.custom_pipeline import Custom_Pipelines - from .basetransform import BaseTransform from .loaders import MultiModalLoader, TextDataLoader @@ -91,6 +89,11 @@ def load(self, name) -> 'Pipeline': Pipeline object. """ + try: + from clarifai_datautils.multimodal.pipeline.custom_pipeline import Custom_Pipelines + except Exception as e: + raise ImportError('cannot Import Custom_Pipelines') from e + self.name = name self.custom_pipelines_map = { 'basic_pdf': Custom_Pipelines.basic_pdf_pipeline(),