Address test issues

Author: Val Brodsky

tests/integration/annotation_import/test_ndjson_validation.py

@@ -1,3 +1,4 @@
+from labelbox.schema.media_type import MediaType
 import pytest
 
 from labelbox import parser
@@ -25,8 +26,11 @@ def get_data_row_id(indx=0):
 
 @pytest.fixture(scope="module", autouse=True)
 def configured_project_with_ontology(client, ontology, rand_gen):
-    project = client.create_project(name=rand_gen(str),
-                                    queue_mode=QueueMode.Batch)
+    project = client.create_project(
+        name=rand_gen(str),
+        queue_mode=QueueMode.Batch,
+        media_type=MediaType.Image,
+    )
     editor = list(
         client.get_labeling_frontends(
             where=LabelingFrontend.name == "editor"))[0]

tests/integration/test_data_rows.py

@@ -3,6 +3,7 @@
 import uuid
 from datetime import datetime
 import json
+from labelbox.schema.media_type import MediaType
 
 import pytest
 import requests
@@ -959,7 +960,8 @@ def test_data_row_bulk_creation_sync_with_same_global_keys(
 def test_create_conversational_text(dataset, conversational_content):
     examples = [
         {
-            **conversational_content, 'media_type': 'CONVERSATIONAL'
+            **conversational_content, 'media_type':
+                MediaType.Conversational.value
         },
         conversational_content,
         {