Skip to content

Commit

Permalink
rebase fix
Browse files Browse the repository at this point in the history
  • Loading branch information
dsikka committed Jul 13, 2023
1 parent 344c27b commit a06dd80
Showing 1 changed file with 0 additions and 42 deletions.
42 changes: 0 additions & 42 deletions tests/deepsparse/pipelines/test_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,48 +104,6 @@ def test_base_pipeline(base_pipeline_example):
assert config.kwargs["base_specific"] == base_pipeline.base_specific


@pytest.fixture
def base_pipeline_example():
@BasePipeline.register(task="base_example")
class BasePipelineExample(BasePipeline):
def __init__(self, base_specific, **kwargs):
self._base_specific = base_specific
super().__init__(**kwargs)

def __call__(self, *args, **kwargs):
pass

def input_schema(self):
pass

def output_schema(self):
pass

@property
def base_specific(self):
return self._base_specific

kwargs = {"alias": "base_alias", "base_specific": "base_specific"}
base_pipeline = BasePipeline.create(task="base_example", **kwargs)
return base_pipeline, BasePipelineExample, kwargs


def test_base_pipeline(base_pipeline_example):
base_pipeline = base_pipeline_example[0]
pipeline = base_pipeline_example[1]
kwargs = base_pipeline_example[-1]

assert base_pipeline.alias == kwargs["alias"]
assert base_pipeline.base_specific == kwargs["base_specific"]

cls = BasePipeline._get_task_constructor("base_example")
assert cls == pipeline

config = base_pipeline.to_config()
assert isinstance(config, PipelineConfig)
assert config.kwargs["base_specific"] == base_pipeline.base_specific


def test_pipeline_executor_num_workers():
executor, _ = _initialize_executor_and_workers(2, None)
assert executor._max_workers == 1
Expand Down

0 comments on commit a06dd80

Please sign in to comment.