From b48888c3e5586c3340cae07c08cba78e62eb992e Mon Sep 17 00:00:00 2001 From: Arne Binder Date: Wed, 6 Mar 2024 17:42:17 +0100 Subject: [PATCH] fix slow test: test_train_predict() --- tests/test_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_predict.py b/tests/test_predict.py index 995ee5c..2c88dd3 100644 --- a/tests/test_predict.py +++ b/tests/test_predict.py @@ -75,7 +75,7 @@ def test_train_predict(tmp_path, cfg_train, cfg_predict): HydraConfig().set_config(cfg_predict) _, object_dict = predict(cfg_predict) - predicted_entities = [list(doc.entities.predictions) for doc in object_dict["documents"]] + predicted_entities = [list(doc.labeled_spans.predictions) for doc in object_dict["documents"]] num_predicted_entities = sum([len(preds) for preds in predicted_entities]) assert num_predicted_entities > 0