diff --git a/tests/integration/test_pipeline_end_to_end.py b/tests/integration/test_pipeline_end_to_end.py index 16bfcd5..c2a7678 100644 --- a/tests/integration/test_pipeline_end_to_end.py +++ b/tests/integration/test_pipeline_end_to_end.py @@ -65,8 +65,8 @@ def test_pipeline_end_to_end(tmp_path, monkeypatch): # run ensure_text_embeddings from pipeline.text_pipeline import ensure_text_embeddings - stored, skipped_existing, skipped_no_text, errors = ensure_text_embeddings( - db_path=db_path, model="test-model" + stored, skipped_existing, skipped_no_text, errors, failed_ids = ( + ensure_text_embeddings(db_path=db_path, model="test-model") ) assert stored == 2