Fix tests

Val Brodsky · Val Brodsky · commit 1dea0e7d2dfa · 2024-06-13T11:53:32.000-07:00
diff --git a/libs/labelbox/src/labelbox/data/annotation_types/label.py b/libs/labelbox/src/labelbox/data/annotation_types/label.py
@@ -19,10 +19,10 @@
 from .video import VideoObjectAnnotation, VideoMaskAnnotation
 from ..ontology import get_feature_schema_lookup
 
-DataType = Union[VideoData, ImageData, TextData, TiledImageData, AudioData,
-                 ConversationData, DicomData, DocumentData, HTMLData,
-                 LlmPromptCreationData, LlmPromptResponseCreationData,
-                 LlmResponseCreationData, GenericDataRowData]
+DataType = Union[GenericDataRowData, VideoData, ImageData, TextData,
+                 TiledImageData, AudioData, ConversationData, DicomData,
+                 DocumentData, HTMLData, LlmPromptCreationData,
+                 LlmPromptResponseCreationData, LlmResponseCreationData]
 
 
 class Label(BaseModel):
