Fix tests

Val Brodsky · Val Brodsky · commit 960968bb23b2 · 2024-10-11T11:26:29.000-07:00
diff --git a/libs/labelbox/src/labelbox/__init__.py b/libs/labelbox/src/labelbox/__init__.py
@@ -26,6 +26,7 @@
     BufferedJsonConverterOutput,
     ExportTask,
     BufferedJsonConverterOutput,
+    StreamType
 )
 from labelbox.schema.iam_integration import IAMIntegration
 from labelbox.schema.identifiable import GlobalKey, UniqueId
diff --git a/libs/labelbox/tests/integration/conftest.py b/libs/labelbox/tests/integration/conftest.py
@@ -34,7 +34,6 @@
 from labelbox.schema.enums import AnnotationImportState
 from labelbox.schema.invite import Invite
 from labelbox.schema.ontology_kind import OntologyKind
-from labelbox.schema.queue_mode import QueueMode
 from labelbox.schema.user import User
 
 

Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,7 @@`
`26`	`26`	`BufferedJsonConverterOutput,`
`27`	`27`	`ExportTask,`
`28`	`28`	`BufferedJsonConverterOutput,`
	`29`	`+ StreamType`
`29`	`30`	`)`
`30`	`31`	`from labelbox.schema.iam_integration import IAMIntegration`
`31`	`32`	`from labelbox.schema.identifiable import GlobalKey, UniqueId`