Merge pull request #425 from Labelbox/develop

3.13.1

Author: msokoloff1

labelbox/__init__.py

@@ -1,5 +1,5 @@
 name = "labelbox"
-__version__ = "3.13.0"
+__version__ = "3.13.1"
 
 from labelbox.schema.project import Project
 from labelbox.client import Client

labelbox/client.py

@@ -11,8 +11,6 @@
 from google.api_core import retry
 import requests
 import requests.exceptions
-from labelbox.data.annotation_types.feature import FeatureSchema
-from labelbox.data.serialization.ndjson.base import DataRow
 
 import labelbox.exceptions
 from labelbox import utils
@@ -621,7 +619,7 @@ def get_roles(self) -> List[Role]:
         """
         return role.get_roles(self)
 
-    def get_data_row(self, data_row_id) -> DataRow:
+    def get_data_row(self, data_row_id):
         """
 
         Returns:
@@ -750,7 +748,7 @@ def get_ontologies(self, name_contains) -> PaginatedCollection:
                                    ['ontologies', 'nodes'], Entity.Ontology,
                                    ['ontologies', 'nextCursor'])
 
-    def get_feature_schema(self, feature_schema_id) -> FeatureSchema:
+    def get_feature_schema(self, feature_schema_id):
         """
         Fetches a feature schema. Only supports top level feature schemas.
 
@@ -867,7 +865,7 @@ def create_ontology(self, name, normalized) -> Ontology:
         res = self.execute(query_str, params)
         return Entity.Ontology(self, res['upsertOntology'])
 
-    def create_feature_schema(self, normalized) -> FeatureSchema:
+    def create_feature_schema(self, normalized):
         """
         Creates a feature schema from normalized data.
             >>> normalized = {'tool': 'polygon',  'name': 'cat', 'color': 'black'}

labelbox/schema/project.py

@@ -13,7 +13,6 @@
 import requests
 
 from labelbox import utils
-from labelbox.data.annotation_types.collection import LabelGenerator
 from labelbox.exceptions import InvalidQueryError, LabelboxError
 from labelbox.orm import query
 from labelbox.orm.db_object import DbObject, Updateable, Deletable
@@ -194,9 +193,7 @@ def export_queued_data_rows(self,
                 self.uid)
             time.sleep(sleep_time)
 
-    def video_label_generator(self,
-                              timeout_seconds=600,
-                              **kwargs) -> LabelGenerator:
+    def video_label_generator(self, timeout_seconds=600, **kwargs):
         """
         Download video annotations
 
@@ -225,7 +222,7 @@ def video_label_generator(self,
                 "Or use project.label_generator() for text and imagery data.")
         return LBV1Converter.deserialize_video(json_data, self.client)
 
-    def label_generator(self, timeout_seconds=600, **kwargs) -> LabelGenerator:
+    def label_generator(self, timeout_seconds=600, **kwargs):
         """
         Download text and image annotations