diff --git a/libs/labelbox/src/labelbox/schema/media_type.py b/libs/labelbox/src/labelbox/schema/media_type.py index 83ba52207..88522c335 100644 --- a/libs/labelbox/src/labelbox/schema/media_type.py +++ b/libs/labelbox/src/labelbox/schema/media_type.py @@ -20,6 +20,8 @@ class MediaType(Enum): Unknown = "UNKNOWN" Unsupported = "UNSUPPORTED" Video = "VIDEO" + Point_Cloud = "POINT_CLOUD" + LLM = "LLM" @classmethod def _missing_(cls, name): diff --git a/libs/labelbox/tests/data/annotation_import/test_data_types.py b/libs/labelbox/tests/data/annotation_import/test_data_types.py index b2ade3730..880c4896d 100644 --- a/libs/labelbox/tests/data/annotation_import/test_data_types.py +++ b/libs/labelbox/tests/data/annotation_import/test_data_types.py @@ -210,8 +210,6 @@ def validate_iso_format(date_string: str): ConversationData, DocumentData, DicomData, - LlmPromptCreationData, - LlmPromptResponseCreationData, LlmResponseCreationData, ], ) diff --git a/libs/labelbox/tests/data/annotation_import/test_generic_data_types.py b/libs/labelbox/tests/data/annotation_import/test_generic_data_types.py index 598a07a93..1ea155fa1 100644 --- a/libs/labelbox/tests/data/annotation_import/test_generic_data_types.py +++ b/libs/labelbox/tests/data/annotation_import/test_generic_data_types.py @@ -154,8 +154,6 @@ def validate_iso_format(date_string: str): ConversationData, DocumentData, DicomData, - LlmPromptCreationData, - LlmPromptResponseCreationData, LlmResponseCreationData, ], ) diff --git a/libs/labelbox/tests/integration/test_project.py b/libs/labelbox/tests/integration/test_project.py index 9343314cb..07a512836 100644 --- a/libs/labelbox/tests/integration/test_project.py +++ b/libs/labelbox/tests/integration/test_project.py @@ -244,7 +244,9 @@ def test_media_type(client, project: Project, rand_gen): for media_type in MediaType.get_supported_members(): # Exclude LLM media types for now, as they are not supported if MediaType[media_type] in [ - MediaType.LLMPromptCreation, MediaType.LLMPromptResponseCreation + MediaType.LLMPromptCreation, + MediaType.LLMPromptResponseCreation, + MediaType.LLM ]: continue