Skip to content

Commit faa5745

Browse files
author
Val Brodsky
committed
Remove unused test fixtures
1 parent c648028 commit faa5745

File tree

3 files changed

+6
-42
lines changed

3 files changed

+6
-42
lines changed

tests/data/assets/ndjson/text_import.json

Lines changed: 0 additions & 25 deletions
This file was deleted.

tests/data/assets/ndjson/text_import_name_only.json

Lines changed: 0 additions & 15 deletions
This file was deleted.

tests/integration/annotation_import/conftest.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -658,7 +658,7 @@ def model_run_with_model_run_data_rows(client, configured_project,
658658
labels = wait_for_label_processing(configured_project)
659659
label_ids = [label.uid for label in labels]
660660
model_run.upsert_labels(label_ids)
661-
time.sleep(3)
661+
time.sleep(300)
662662
yield model_run
663663
model_run.delete()
664664
# TODO: Delete resources when that is possible ..
@@ -670,6 +670,11 @@ def model_run_with_all_project_labels(client, configured_project,
670670
wait_for_label_processing):
671671
configured_project.enable_model_assisted_labeling()
672672

673+
data_row_ids = configured_project.data_row_ids
674+
675+
configured_project._wait_until_data_rows_are_processed(
676+
data_row_ids=data_row_ids)
677+
673678
upload_task = LabelImport.create_from_objects(
674679
client, configured_project.uid, f"label-import-{uuid.uuid4()}",
675680
model_run_predictions)
@@ -680,7 +685,6 @@ def model_run_with_all_project_labels(client, configured_project,
680685
) == 0, f"Label Import {upload_task.name} failed with errors {upload_task.errors}"
681686
wait_for_label_processing(configured_project)
682687
model_run.upsert_labels(project_id=configured_project.uid)
683-
time.sleep(3)
684688
yield model_run
685689
model_run.delete()
686690
# TODO: Delete resources when that is possible ..

0 commit comments

Comments
 (0)