diff --git a/datalabeling/label_image_test.py b/datalabeling/label_image_test.py index 3917ef0481a..ddb74a03105 100644 --- a/datalabeling/label_image_test.py +++ b/datalabeling/label_image_test.py @@ -88,12 +88,15 @@ def instruction(): # to be deleted before the annotation_spec_set can be deleted. @pytest.mark.slow def test_label_image(capsys, annotation_spec_set, instruction, dataset): + annotation_spec_set_name = annotation_spec_set.name + instruction_name = instruction.name + dataset_name = dataset.name # Start labeling. response = label_image.label_image( - dataset.name, - instruction.name, - annotation_spec_set.name + dataset_name, + instruction_name, + annotation_spec_set_name ) out, _ = capsys.readouterr() assert 'Label_image operation name: ' in out diff --git a/datalabeling/label_text_test.py b/datalabeling/label_text_test.py index 4894c38304c..0e1f37d0c01 100644 --- a/datalabeling/label_text_test.py +++ b/datalabeling/label_text_test.py @@ -88,12 +88,15 @@ def instruction(): # to be deleted before the annotation_spec_set can be deleted. @pytest.mark.slow def test_label_text(capsys, annotation_spec_set, instruction, dataset): + annotation_spec_set_name = annotation_spec_set.name + instruction_name = instruction.name + dataset_name = dataset.name # Start labeling. response = label_text.label_text( - dataset.name, - instruction.name, - annotation_spec_set.name + dataset_name, + instruction_name, + annotation_spec_set_name ) out, _ = capsys.readouterr() assert 'Label_text operation name: ' in out diff --git a/datalabeling/label_video_test.py b/datalabeling/label_video_test.py index 606b28c1c9a..fd466b052c8 100644 --- a/datalabeling/label_video_test.py +++ b/datalabeling/label_video_test.py @@ -88,12 +88,15 @@ def instruction(): # to be deleted before the annotation_spec_set can be deleted. @pytest.mark.slow def test_label_video(capsys, annotation_spec_set, instruction, dataset): + annotation_spec_set_name = annotation_spec_set.name + instruction_name = instruction.name + dataset_name = dataset.name # Start labeling. response = label_video.label_video( - dataset.name, - instruction.name, - annotation_spec_set.name + dataset_name, + instruction_name, + annotation_spec_set_name ) out, _ = capsys.readouterr() assert 'Label_video operation name: ' in out