adjust for pie-modules

ArneBinder · Nov 27, 2023 · e1f273d · e1f273d
1 parent 9e5da9b
commit e1f273d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dataset_builders/pie/argmicro/requirements.txt b/dataset_builders/pie/argmicro/requirements.txt
@@ -1 +1 @@
-pie-datasets>=0.3.3,<0.6.0
+pie-datasets>=0.3.3,<0.7.0
diff --git a/tests/dataset_builders/pie/test_argmicro.py b/tests/dataset_builders/pie/test_argmicro.py
@@ -4,7 +4,7 @@
 
 import pytest
 from datasets import disable_caching, load_dataset
-from pie_models.document.processing import tokenize_document
+from pie_modules.document.processing import tokenize_document
 from pytorch_ie.core import Document
 from pytorch_ie.documents import TextDocumentWithLabeledSpansAndBinaryRelations
 from transformers import AutoTokenizer, PreTrainedTokenizer
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		pie-datasets>=0.3.3,<0.6.0
		pie-datasets>=0.3.3,<0.7.0