Merge pull request #300 from Labelbox/ms/fix-import

fix circular import

Author: msokoloff1

CHANGELOG.md

@@ -1,4 +1,7 @@
 # Changelog
+# Version 3.6.1 (2021-04-10)
+* Fix import error that appears when exporting labels
+
 # Version 3.6.0 (2021-04-10)
 ## Added
 * Bulk export metadata with `DataRowMetadataOntology.bulk_export()`

labelbox/__init__.py

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

labelbox/data/annotation_types/label.py

@@ -3,6 +3,7 @@
 
 from pydantic import BaseModel, validator
 
+import labelbox
 from labelbox.schema import ontology
 from .annotation import (ClassificationAnnotation, ObjectAnnotation,
                          VideoClassificationAnnotation, VideoObjectAnnotation)
@@ -12,7 +13,6 @@
 from .metrics import ScalarMetric, ConfusionMatrixMetric
 from .types import Cuid
 from ..ontology import get_feature_schema_lookup
-from ... import Dataset
 
 
 class Label(BaseModel):
@@ -103,7 +103,7 @@ def add_url_to_masks(self, signer) -> "Label":
             mask.create_url(signer)
         return self
 
-    def create_data_row(self, dataset: "Dataset",
+    def create_data_row(self, dataset: "labelbox.Dataset",
                         signer: Callable[[bytes], str]) -> "Label":
         """
         Creates a data row and adds to the given dataset.