update pie-datasets to 0.4.0 (#147)

* update pie-datasets to 0.4.0 * fix test_train_predict
ArneBinder · Nov 23, 2023 · 8774467 · 8774467
1 parent c6afd2b
commit 8774467
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/configs/dataset/conll2003.yaml b/configs/dataset/conll2003.yaml
@@ -1,4 +1,4 @@
 _target_: pie_datasets.DatasetDict.load_dataset
 
 path: pie/conll2003
-revision: 0fa8689b44ca9885b77276205a7dab3b562266b9
+revision: 95ee94f78010d00cb5a94c38ebbddd07273be8c0
diff --git a/configs/dataset/conll2003_base.yaml b/configs/dataset/conll2003_base.yaml
@@ -5,4 +5,4 @@ _target_: src.utils.execute_pipeline
 input:
   _target_: pie_datasets.DatasetDict.load_dataset
   path: pie/conll2003
-  revision: 0fa8689b44ca9885b77276205a7dab3b562266b9
+  revision: 95ee94f78010d00cb5a94c38ebbddd07273be8c0
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 # --------- pytorch-ie --------- #
 pytorch-ie>=0.28.0,<0.30.0
-pie-datasets>=0.3.1,<0.4.0
+pie-datasets>=0.3.3,<0.5.0
 
 # --------- hydra --------- #
 hydra-core>=1.3.0

diff --git 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.labeled_spans.predictions) for doc in object_dict["documents"]]
+    predicted_entities = [list(doc.entities.predictions) for doc in object_dict["documents"]]
     num_predicted_entities = sum([len(preds) for preds in predicted_entities])
     assert num_predicted_entities > 0