diff --git a/libs/labelbox/tests/data/annotation_import/test_model_run.py b/libs/labelbox/tests/data/annotation_import/test_model_run.py index 3b2e04e62..b267bcf6f 100644 --- a/libs/labelbox/tests/data/annotation_import/test_model_run.py +++ b/libs/labelbox/tests/data/annotation_import/test_model_run.py @@ -168,16 +168,14 @@ def get_model_run_status(): errorMessage) -def test_model_run_split_assignment_by_data_row_ids( - model_run, dataset, image_url, configured_project_with_one_data_row): +def test_model_run_split_assignment_by_data_row_ids(model_run, dataset, + image_url): n_data_rows = 2 data_rows = dataset.create_data_rows([{ "row_data": image_url } for _ in range(n_data_rows)]) data_rows.wait_till_done() data_row_ids = [data_row['id'] for data_row in data_rows.result] - configured_project_with_one_data_row._wait_until_data_rows_are_processed( - data_row_ids=data_row_ids) model_run.upsert_data_rows(data_row_ids) with pytest.raises(ValueError): diff --git a/libs/labelbox/tests/integration/test_filtering.py b/libs/labelbox/tests/integration/test_filtering.py index f0558f04c..082809935 100644 --- a/libs/labelbox/tests/integration/test_filtering.py +++ b/libs/labelbox/tests/integration/test_filtering.py @@ -26,7 +26,8 @@ def project_to_test_where(client, rand_gen): # other builds simultaneously adding projects to test org def test_where(client, project_to_test_where): p_a, p_b, p_c = project_to_test_where - p_a_name, p_b_name = [p.name for p in [p_a, p_b]] + p_a_name = p_a.name + p_b_name = p_b.name def get(where=None): date_where = Project.created_at >= p_a.created_at diff --git a/libs/labelbox/tests/integration/test_task.py b/libs/labelbox/tests/integration/test_task.py index 07309928a..af0ce9f4a 100644 --- a/libs/labelbox/tests/integration/test_task.py +++ b/libs/labelbox/tests/integration/test_task.py @@ -67,7 +67,7 @@ def test_task_success_label_export(client, configured_project_with_label): user = client.get_user() task = None for task in user.created_tasks(): - if task.name != 'JSON Import' and task.type != 'adv-upsert-data-rows': + if task.name != 'JSON Import' and task.type != 'adv-upsert-data-rows' and task.type != 'mea-label-registration': break with pytest.raises(ValueError) as exc_info: