Update MediaType methods' names, add media_type import

Author: paulnoirel

labelbox/__init__.py

@@ -26,3 +26,4 @@
 from labelbox.schema.iam_integration import IAMIntegration
 from labelbox.schema.resource_tag import ResourceTag
 from labelbox.schema.project_resource_tag import ProjectResourceTag
+from labelbox.schema.media_type import MediaType

labelbox/client.py

@@ -615,11 +615,11 @@ def create_project(self, **kwargs) -> Project:
         """
         media_type = kwargs.get("media_type")
         if media_type:
-            if MediaType.is_accepted(media_type):
+            if MediaType.is_supported(media_type):
                 kwargs["media_type"] = media_type.value
             else:
                 raise TypeError(f"{media_type} is not a valid media type. Use"
-                                f" any of {MediaType.get_accepted_members()}"
+                                f" any of {MediaType.get_supported_members()}"
                                 " from MediaType. Example: MediaType.Image.")
 
         return self._create(Entity.Project, kwargs)

labelbox/schema/__init__.py

@@ -20,3 +20,4 @@
 import labelbox.schema.data_row_metadata
 import labelbox.schema.batch
 import labelbox.schema.iam_integration
+import labelbox.schema.media_type

labelbox/schema/media_type.py

@@ -34,12 +34,12 @@ def _missing_(cls, name):
                 return member
 
     @classmethod
-    def is_accepted(cls, value):
+    def is_supported(cls, value):
         return isinstance(value,
                           cls) and value not in [cls.Unknown, cls.Unsupported]
 
     @classmethod
-    def get_accepted_members(cls):
+    def get_supported_members(cls):
         return [
             item for item in cls.__members__
             if item not in ["Unknown", "Unsupported"]

labelbox/schema/project.py

@@ -100,11 +100,11 @@ def update(self, **kwargs):
 
         media_type = kwargs.get("media_type")
         if media_type:
-            if MediaType.is_accepted(media_type):
+            if MediaType.is_supported(media_type):
                 kwargs["media_type"] = media_type.value
             else:
                 raise TypeError(f"{media_type} is not a valid media type. Use"
-                                f" any of {MediaType.get_accepted_members()}"
+                                f" any of {MediaType.get_supported_members()}"
                                 " from MediaType. Example: MediaType.Image.")
 
         return super().update(**kwargs)

tests/integration/test_project.py

@@ -214,7 +214,7 @@ def test_media_type(client, configured_project: Project, rand_gen):
     assert project.media_type == MediaType.Image
     project.delete()
 
-    for media_type in MediaType.get_accepted_members():
+    for media_type in MediaType.get_supported_members():
 
         project = client.create_project(name=rand_gen(str),
                                         media_type=MediaType[media_type])