Skip to content

Commit a1c1d54

Browse files
rmitschRaphael Mitsch
andauthored
test: Revert test changes (#215)
* fix: Attempt to fix CI tests. * chore: Add debug logging. * chore: Add debug logging. * chore: Add debug logging. * chore: Add debug logging. * chore: Add debug logging. * fix: Robust GliNER result processing. * fix: Robust GliNER result processing. * fix: Robust GliNER result processing. * fix: Robust GliNER result processing. * fix: Robust GliNER result processing. * fix: Robust GliNER result processing. * fix: Robust GliNER result processing. * chore: Revert NER test changes. * chore: Revert classification test changes. --------- Co-authored-by: Raphael Mitsch <raphael@climatiq.com>
1 parent acf0441 commit a1c1d54

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sieves/tests/tasks/predictive/test_classification.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def _run(
6666
if test_hf_conversion:
6767
_to_hf_dataset(task, docs, multilabel)
6868

69-
@pytest.mark.parametrize("batch_runtime", (EngineType.gliner,), indirect=["batch_runtime"])
69+
@pytest.mark.parametrize("batch_runtime", Classification.supports(), indirect=["batch_runtime"])
7070
@pytest.mark.parametrize("fewshot", [True, False])
7171
@pytest.mark.parametrize("multilabel", [True, False])
7272
def test_run(classification_docs, batch_runtime, fewshot, multilabel):

0 commit comments

Comments
 (0)