3.6 compatibility

Author: Matt Sokoloff

labelbox/data/annotation_types/types.py

@@ -1,3 +1,5 @@
+import sys
+import logging
 from typing import Generic, TypeVar
 from typing_extensions import Annotated
 
@@ -9,6 +11,8 @@
 
 DType = TypeVar('DType')
 
+logger = logging.getLogger(__name__)
+
 
 class TypedArray(np.ndarray, Generic[DType]):
 
@@ -20,8 +24,12 @@ def __get_validators__(cls):
     def validate(cls, val, field: ModelField):
         if not isinstance(val, np.ndarray):
             raise TypeError(f"Expected numpy array. Found {type(val)}")
-        dtype_field = field.sub_fields[0]
-        actual_dtype = dtype_field.type_.__args__[0]
+
+        if sys.version_info.minor > 6:
+            actual_dtype = field.sub_fields[0].type_.__args__[0]
+        else:
+            actual_dtype = field.sub_fields[0].type_.__values__[0]
+
         if val.dtype != actual_dtype:
             raise TypeError(
                 f"Expected numpy array have type {actual_dtype}. Found {val.dtype}"

setup.py

@@ -22,13 +22,16 @@
     install_requires=[
         "backoff==1.10.0",
         "backports-datetime-fromisoformat==1.0.0; python_version < '3.7.0'",
-        "dataclasses==0.7; python_version < '3.7.0'", "ndjson==0.3.1",
-        "requests>=2.22.0", "google-api-core>=1.22.1", "pydantic>=1.8,<2.0",
+        "dataclasses==0.7; python_version < '3.7.0'",
+        "ndjson==0.3.1",
+        "requests>=2.22.0",
+        "google-api-core>=1.22.1",
+        "pydantic>=1.8,<2.0",
     ],
     extras_require={
         'data': [
-            "shapely", "geojson", "numpy", "rasterio", "PILLOW", "opencv-python",
-        "typeguard", "tqdm"
+            "shapely", "geojson", "numpy", "rasterio", "PILLOW",
+            "opencv-python", "typeguard", "tqdm"
         ],
     },
     classifiers=[