address comments

Author: rllin

labelbox/exceptions.py

@@ -88,3 +88,8 @@ class ApiLimitError(LabelboxError):
 class MalformedQueryException(Exception):
     """ Raised when the user submits a malformed query."""
     pass
+
+
+class UuidError(LabelboxError):
+    """ Raised when there are repeat Uuid's in bulk import request."""
+    pass

labelbox/schema/bulk_import_request.py

@@ -279,13 +279,6 @@ def create_from_local_file(cls,
         return cls(client, response_data["createBulkImportRequest"])
 
 
-class NdjsonError(Exception):
-    pass
-
-
-class UuidError(NdjsonError):
-    pass
-
 
 def _validate_ndjson(lines: Iterable[Dict[str, Any]]) -> None:
     """Validate individual ndjson lines.
@@ -297,6 +290,7 @@ def _validate_ndjson(lines: Iterable[Dict[str, Any]]) -> None:
     for line in lines:
         uuid = line['uuid']
         if uuid in uuids:
-            raise UuidError(f'{uuid} already used in this import job, '
-                            'must be unique for the project.')
-        uuids.add(line['uuid'])
+            raise labelbox.exceptions.UuidError(
+                f'{uuid} already used in this import job, '
+                'must be unique for the project.')
+        uuids.add(uuid)