Skip to content

Commit

Permalink
edit per early feedback
Browse files Browse the repository at this point in the history
  • Loading branch information
idalr committed Nov 3, 2023
1 parent 89c988b commit e8899dd
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 5 deletions.
File renamed without changes.
File renamed without changes.
9 changes: 4 additions & 5 deletions tests/dataset_builders/pie/test_cdcp.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,17 +19,16 @@
document_to_example,
example_to_document,
)
from src.document.types import TokenDocumentWithLabeledSpansAndBinaryRelations
from pie_datasets.document.types import TokenDocumentWithLabeledSpansAndBinaryRelations
from tests import FIXTURES_ROOT
from tests.dataset_builders.common import _deep_compare
from tests.dataset_builders.common import PIE_BASE_PATH, _deep_compare

disable_caching()

DATASET_NAME = "cdcp"
SPLIT_SIZES = {"train": 581, "test": 150}
# HF_DATASET_PATH = "DFKI-SLT/cdcp"
HF_DATASET_PATH = CDCP.BASE_DATASET_PATH
PIE_DATASET_PATH = "pie/cdcp"
PIE_DATASET_PATH = PIE_BASE_PATH / DATASET_NAME # "pie/cdcp"
DATA_PATH = FIXTURES_ROOT / "dataset_builders" / "cdcp_acl17.zip"

HF_EXAMPLE_00195 = {
Expand Down Expand Up @@ -171,7 +170,7 @@ def test_example_to_document_and_back_all(

@pytest.fixture(scope="module")
def dataset() -> DatasetDict:
return DatasetDict.load_dataset(PIE_DATASET_PATH)
return DatasetDict.load_dataset(str(PIE_DATASET_PATH))


def test_pie_dataset(dataset):
Expand Down

0 comments on commit e8899dd

Please sign in to comment.