From 04d6ae208ea58e3d7933be253611349475545e27 Mon Sep 17 00:00:00 2001 From: Dipika Sikka Date: Tue, 5 Dec 2023 15:59:41 +0000 Subject: [PATCH 1/2] add pipeline create method for pipeline creation using the operator registry --- src/deepsparse/v2/pipeline.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/deepsparse/v2/pipeline.py b/src/deepsparse/v2/pipeline.py index 450a6702c4..98273e8f0f 100644 --- a/src/deepsparse/v2/pipeline.py +++ b/src/deepsparse/v2/pipeline.py @@ -173,6 +173,15 @@ def _apply_split(self, inp: Any, inference_state: InferenceState): ) return self.condense_inputs(outputs) + @staticmethod + def create(task: str, **kwargs) -> "Pipeline": + """ + :param task: Pipeline task + :param kwargs: extra task specific kwargs to be passed to the Pipeline + :return: pipeline object initialized for the given task + """ + return Operator.create(task=task, **kwargs) + def run( self, *args, From 78f4d7fd5ce1988a7cf4be7bb570e12f90a28089 Mon Sep 17 00:00:00 2001 From: Dipika Sikka Date: Tue, 5 Dec 2023 17:22:31 +0000 Subject: [PATCH 2/2] add instance check --- src/deepsparse/v2/pipeline.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/deepsparse/v2/pipeline.py b/src/deepsparse/v2/pipeline.py index 98273e8f0f..ff9014799b 100644 --- a/src/deepsparse/v2/pipeline.py +++ b/src/deepsparse/v2/pipeline.py @@ -180,7 +180,13 @@ def create(task: str, **kwargs) -> "Pipeline": :param kwargs: extra task specific kwargs to be passed to the Pipeline :return: pipeline object initialized for the given task """ - return Operator.create(task=task, **kwargs) + pipeline = Operator.create(task=task, **kwargs) + if not isinstance(pipeline, Pipeline): + raise RuntimeError( + "Pipeline was not created for the given task. The " + "provided task should be registered using the OperatorRegistry" + ) + return pipeline def run( self,