Skip to content

Commit 6133f89

Browse files
committed
add test_pipeline_with_dataset_never_cached
1 parent a983f7c commit 6133f89

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tests/test_pipeline.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,3 +145,16 @@ def test_pipeline_with_dataset(dataset, prepared_taskmodule, mock_model, inplace
145145
assert not (id(returned_document) == id(document))
146146
assert not document.entities.predictions
147147
assert returned_document.entities.predictions
148+
149+
150+
@pytest.mark.slow
151+
def test_pipeline_with_dataset_never_cached(dataset, prepared_taskmodule, mock_model):
152+
153+
train_dataset = dataset["train"]
154+
155+
pipeline = Pipeline(model=mock_model, taskmodule=prepared_taskmodule, device=-1, inplace=False)
156+
157+
returned_documents1 = pipeline(train_dataset, predict_field="entities")
158+
returned_documents2 = pipeline(train_dataset, predict_field="entities")
159+
160+
assert returned_documents1._fingerprint != returned_documents2._fingerprint

0 commit comments

Comments
 (0)