[PLT-1485] Removed Datatypes

libs/labelbox/src/labelbox/data/annotation_types/__init__.py

@@ -32,18 +32,8 @@
 from .classification import Radio
 from .classification import Text
 
-from .data import AudioData
-from .data import ConversationData
-from .data import DicomData
-from .data import DocumentData
-from .data import HTMLData
-from .data import ImageData
+from .data import GenericDataRowData
 from .data import MaskData
-from .data import TextData
-from .data import VideoData
-from .data import LlmPromptResponseCreationData
-from .data import LlmPromptCreationData
-from .data import LlmResponseCreationData
 
 from .label import Label
 from .collection import LabelGenerator

libs/labelbox/src/labelbox/data/annotation_types/collection.py

@@ -40,47 +40,6 @@ def _assign_ids(label: Label):
         self._fns["assign_feature_schema_ids"] = _assign_ids
         return self
 
-    def add_url_to_data(
-        self, signer: Callable[[bytes], str]
-    ) -> "LabelGenerator":
-        """
-        Creates signed urls for the data
-        Only uploads url if one doesn't already exist.
-
-        Args:
-            signer: A function that accepts bytes and returns a signed url.
-        Returns:
-            LabelGenerator that signs urls as data is accessed
-        """
-
-        def _add_url_to_data(label: Label):
-            label.add_url_to_data(signer)
-            return label
-
-        self._fns["add_url_to_data"] = _add_url_to_data
-        return self
-
-    def add_to_dataset(
-        self, dataset: "Entity.Dataset", signer: Callable[[bytes], str]
-    ) -> "LabelGenerator":
-        """
-        Creates data rows from each labels data object and attaches the data to the given dataset.
-        Updates the label's data object to have the same external_id and uid as the data row.
-
-        Args:
-            dataset: labelbox dataset object to add the new data row to
-            signer: A function that accepts bytes and returns a signed url.
-        Returns:
-            LabelGenerator that updates references to the new data rows as data is accessed
-        """
-
-        def _add_to_dataset(label: Label):
-            label.create_data_row(dataset, signer)
-            return label
-
-        self._fns["assign_datarow_ids"] = _add_to_dataset
-        return self
-
     def add_url_to_masks(
         self, signer: Callable[[bytes], str]
     ) -> "LabelGenerator":

libs/labelbox/src/labelbox/data/annotation_types/data/__init__.py

@@ -1,12 +1,2 @@
-from .audio import AudioData
-from .conversation import ConversationData
-from .dicom import DicomData
-from .document import DocumentData
-from .html import HTMLData
-from .raster import ImageData
 from .raster import MaskData
-from .text import TextData
-from .video import VideoData
-from .llm_prompt_response_creation import LlmPromptResponseCreationData
-from .llm_prompt_creation import LlmPromptCreationData
-from .llm_response_creation import LlmResponseCreationData
+from .generic_data_row_data import GenericDataRowData

libs/labelbox/src/labelbox/data/annotation_types/data/raster.py

@@ -11,8 +11,9 @@
 from requests.exceptions import ConnectTimeout
 from typing_extensions import Literal
 
+from pydantic import BaseModel, model_validator, ConfigDict
+
 from ..types import TypedArray
-from .base_data import BaseData
 
 
 class RasterData(BaseModel, ABC):
@@ -222,6 +223,3 @@ class MaskData(RasterData):
         url: Optional[str] = None
         arr: Optional[TypedArray[Literal['uint8']]] = None
     """
-
-
-class ImageData(RasterData, BaseData): ...