Skip to content

Commit 448164d

Browse files
author
Val Brodsky
committed
Add explanation on how to supply data row ids to prediction_id_mapping
1 parent 873b659 commit 448164d

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

tests/integration/annotation_import/conftest.py

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -505,13 +505,11 @@ def configured_project(client, initial_dataset, ontology, rand_gen, image_url):
505505
client.get_labeling_frontends(
506506
where=LabelingFrontend.name == "editor"))[0]
507507
project.setup(editor, ontology)
508-
num_rows = 0
509508

510509
data_row_ids = []
511510

512511
for _ in range(len(ontology['tools']) + len(ontology['classifications'])):
513512
data_row_ids.append(dataset.create_data_row(row_data=image_url).uid)
514-
num_rows += 1
515513
project._wait_until_data_rows_are_processed(data_row_ids=data_row_ids,
516514
sleep_interval=3)
517515

@@ -605,6 +603,22 @@ def configured_project_with_one_data_row(client, ontology, rand_gen,
605603
# At the moment it expects only one feature per tool type and this creates unnecessary coupling between differet tests
606604
# In an example of a 'rectangle' we have extended to support multiple instances of the same tool type
607605
# TODO: we will support this approach in the future for all tools
606+
#
607+
"""
608+
Please note that this fixture now offers the flexibility to configure three different strategies for generating data row ids for predictions:
609+
Default(configured_project fixture):
610+
configured_project that generates a data row for each member of ontology.
611+
This makes sure each prediction has its own data row id. This is applicable to prediction upload cases when last label overwrites existing ones
612+
613+
Optimized Strategy (configured_project_with_one_data_row fixture):
614+
This fixture has only one data row and all predictions will be mapped to it
615+
616+
Custom Data Row IDs Strategy:
617+
Individuals can create their own fixture to supply data row ids.
618+
This particular fixture, termed "hardcoded_datarow_id," should be defined locally within a test file.
619+
"""
620+
621+
608622
@pytest.fixture
609623
def prediction_id_mapping(ontology, request):
610624
# Maps tool types to feature schema ids

0 commit comments

Comments
 (0)