scale-nucleus 0.16.13__py3-none-any.whl → 0.16.14__py3-none-any.whl

nucleus/__init__.py

@@ -45,13 +45,17 @@ __all__ = [
 import datetime
 import os
 import warnings
-from typing import Any, Dict, List, Optional, Tuple, Union
-
-try:
-    # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
-    import pydantic.v1 as pydantic
-except ImportError:
-    import pydantic
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
+
+if TYPE_CHECKING:
+    # Backwards compatibility is even uglier with mypy
+    from pydantic.v1 import parse_obj_as
+else:
+    try:
+        # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+        from pydantic.v1 import parse_obj_as
+    except ImportError:
+        from pydantic import parse_obj_as
 
 import requests
 import tqdm
@@ -216,7 +220,11 @@ class NucleusClient:
             List of all datasets accessible to user
         """
         response = self.make_request({}, "dataset/details", requests.get)
-        dataset_details = pydantic.parse_obj_as(List[DatasetDetails], response)
+        dataset_details = (
+            parse_obj_as(  # pylint: disable=used-before-assignment
+                List[DatasetDetails], response
+            )
+        )
         return [
             Dataset(d.id, client=self, name=d.name) for d in dataset_details
         ]

nucleus/data_transfer_object/job_status.py

@@ -1,15 +1,19 @@
 # pylint: disable=E0213
 
 from datetime import datetime
-from typing import List, Optional, Union
+from typing import TYPE_CHECKING, List, Optional, Union
 
 from dateutil.parser import ParserError, parse
 
-try:
-    # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+if TYPE_CHECKING:
+    # Backwards compatibility is even uglier with mypy
     from pydantic.v1 import validator
-except ImportError:
-    from pydantic import validator
+else:
+    try:
+        # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+        from pydantic.v1 import validator
+    except ImportError:
+        from pydantic import validator
 
 from nucleus.constants import JOB_REQ_LIMIT
 from nucleus.job import CustomerJobTypes
@@ -24,7 +28,9 @@ class JobInfoRequestPayload(ImmutableModel):
     limit: Optional[int]
     show_completed: bool
 
-    @validator("from_date", "to_date")
+    @validator(  # pylint: disable=used-before-assignment
+        "from_date", "to_date"
+    )
     def ensure_date_format(cls, date):
         if date is None:
             return None

nucleus/pydantic_base.py

@@ -4,14 +4,20 @@ We started using pydantic during v1 and are kind of stuck with it now unless we
 As a library we want to support v1 and v2 such that we're not causing downstream problems for our users.
 This means we have to do some import shenanigans to support both v1 and v2.
 """
-try:
-    # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
-    from pydantic.v1 import BaseModel  # pylint: disable=no-name-in-module
-except ImportError:
-    from pydantic import BaseModel
+from typing import TYPE_CHECKING
 
+if TYPE_CHECKING:
+    # Backwards compatibility is even uglier with mypy
+    from pydantic.v1 import BaseModel, Extra, ValidationError
+else:
+    try:
+        # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+        from pydantic.v1 import BaseModel  # pylint: disable=no-name-in-module
+    except ImportError:
+        from pydantic import BaseModel
 
-class ImmutableModel(BaseModel):
+
+class ImmutableModel(BaseModel):  # pylint: disable=used-before-assignment
     class Config:
         allow_mutation = False
 

nucleus/test_launch_integration.py

@@ -1,24 +1,28 @@
 import io
-from typing import Any, Callable, Dict, List, Optional, Type
+from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Type
 
 from PIL import Image, ImageDraw
 
-try:
-    # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+if TYPE_CHECKING:
+    # Backwards compatibility is even uglier with mypy
     from pydantic.v1 import BaseModel, Extra, ValidationError
-except ImportError:
-    from pydantic import BaseModel, Extra, ValidationError
+else:
+    try:
+        # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+        from pydantic.v1 import BaseModel, Extra, ValidationError
+    except ImportError:
+        from pydantic import BaseModel, Extra, ValidationError
 
 # From scaleapi/server/src/lib/select/api/types.ts
 # These classes specify how user models must pass output to Launch + Nucleus.
 
 
-class PointModel(BaseModel):
+class PointModel(BaseModel):  # pylint: disable=used-before-assignment
     x: float
     y: float
 
     class Config:
-        extra = Extra.forbid
+        extra = Extra.forbid  # pylint: disable=used-before-assignment
 
 
 class BoxGeometryModel(BaseModel):
@@ -106,7 +110,7 @@ def verify_output(
     for annotation in annotation_list:
         try:
             model.parse_obj(annotation)
-        except ValidationError as e:
+        except ValidationError as e:  # pylint: disable=used-before-assignment
             raise ValueError("Failed validation") from e
         if annotation["type"] != annotation_type:
             raise ValueError(

nucleus/utils.py

@@ -363,7 +363,7 @@ def upload_to_presigned_url(presigned_url: str, file_pointer: IO):
     # TODO optimize this further to deal with truly huge files and flaky internet connection.
     upload_response = requests.put(presigned_url, file_pointer)
     if not upload_response.ok:
-        raise HTTPError(
+        raise HTTPError(  # type: ignore
             f"Tried to put a file to url, but failed with status {upload_response.status_code}. The detailed error was: {upload_response.text}"
         )
 

nucleus/validate/data_transfer_objects/eval_function.py

@@ -1,10 +1,14 @@
-from typing import Any, Dict, List, Optional
+from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
-try:
-    # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+if TYPE_CHECKING:
+    # Backwards compatibility is even uglier with mypy
     from pydantic.v1 import validator
-except ImportError:
-    from pydantic import validator
+else:
+    try:
+        # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+        from pydantic.v1 import validator
+    except ImportError:
+        from pydantic import validator
 
 from ...pydantic_base import ImmutableModel
 from ..constants import ThresholdComparison
@@ -63,7 +67,7 @@ class EvaluationCriterion(ImmutableModel):
     threshold: float
     eval_func_arguments: Dict[str, Any]
 
-    @validator("eval_function_id")
+    @validator("eval_function_id")  # pylint: disable=used-before-assignment
     def valid_eval_function_id(cls, v):  # pylint: disable=no-self-argument
         if not v.startswith("ef_"):
             raise ValueError(f"Expected field to start with 'ef_', got '{v}'")

nucleus/validate/data_transfer_objects/scenario_test.py

@@ -1,10 +1,14 @@
-from typing import List
+from typing import TYPE_CHECKING, List
 
-try:
-    # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+if TYPE_CHECKING:
+    # Backwards compatibility is even uglier with mypy
     from pydantic.v1 import validator
-except ImportError:
-    from pydantic import validator
+else:
+    try:
+        # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+        from pydantic.v1 import validator
+    except ImportError:
+        from pydantic import validator
 
 from nucleus.pydantic_base import ImmutableModel
 
@@ -19,7 +23,7 @@ class CreateScenarioTestRequest(ImmutableModel):
     slice_id: str
     evaluation_functions: List[EvalFunctionListEntry]
 
-    @validator("slice_id")
+    @validator("slice_id")  # pylint: disable=used-before-assignment
     def startswith_slice_indicator(cls, v):  # pylint: disable=no-self-argument
         if not v.startswith("slc_"):
             raise ValueError(f"Expected field to start with 'slc_', got '{v}'")

nucleus/validate/data_transfer_objects/scenario_test_evaluations.py

@@ -1,10 +1,14 @@
-from typing import Optional
+from typing import TYPE_CHECKING, Optional
 
-try:
-    # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+if TYPE_CHECKING:
+    # Backwards compatibility is even uglier with mypy
     from pydantic.v1 import root_validator, validator
-except ImportError:
-    from pydantic import root_validator, validator
+else:
+    try:
+        # NOTE: we always use pydantic v1 but have to do these shenanigans to support both v1 and v2
+        from pydantic.v1 import root_validator, validator
+    except ImportError:
+        from pydantic import root_validator, validator
 
 from nucleus.pydantic_base import ImmutableModel
 
@@ -16,7 +20,7 @@ class EvaluationResult(ImmutableModel):
     score: float = 0
     weight: float = 1
 
-    @root_validator()
+    @root_validator()  # pylint: disable=used-before-assignment
     def is_item_or_scene_provided(
         cls, values
     ):  # pylint: disable=no-self-argument
@@ -31,7 +35,7 @@ class EvaluationResult(ImmutableModel):
             )
         return values
 
-    @validator("weight")
+    @validator("weight")  # pylint: disable=used-before-assignment
     def is_normalized(cls, v):  # pylint: disable=no-self-argument
         if 0 <= v <= 1:
             return v

{scale_nucleus-0.16.13.dist-info → scale_nucleus-0.16.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scale-nucleus
-Version: 0.16.13
+Version: 0.16.14
 Summary: The official Python client library for Nucleus, the Data Platform for AI
 Home-page: https://scale.com/nucleus
 License: MIT
@@ -24,7 +24,7 @@ Requires-Dist: click (>=7.1.2,<9.0)
 Requires-Dist: nest-asyncio (>=1.5.1,<2.0.0)
 Requires-Dist: numpy (>=1.19.5) ; python_version >= "3.7" and python_version < "3.10"
 Requires-Dist: numpy (>=1.22.0) ; python_version >= "3.10"
-Requires-Dist: pydantic (>=1.8.2,<2.0.0)
+Requires-Dist: pydantic (>=1.8.2)
 Requires-Dist: python-dateutil (>=2.8.2,<3.0.0)
 Requires-Dist: questionary (>=1.10.0,<2.0.0)
 Requires-Dist: rasterio (>=1.2.0) ; extra == "metrics"

{scale_nucleus-0.16.13.dist-info → scale_nucleus-0.16.14.dist-info}/RECORD

@@ -10,7 +10,7 @@ cli/nu.py,sha256=0f71zPq4fe3I1ghhiSRQi39ENhAzoLPdhz_vh8vxSI8,2074
 cli/reference.py,sha256=RuHVhmGTZNe0MfwpL96YjJdaH0OJzg98rz4xeIu4hJU,256
 cli/slices.py,sha256=nxq_Zg1m5oXuhz0ibyHkElvyVWt1AcE9tG-fN4CQxF8,1397
 cli/tests.py,sha256=NiwEVGuF08_jlCiKEIjKhwq55NvyU4xvPEJW5MJmdZg,4590
-nucleus/__init__.py,sha256=mpa_n1DRo_2_P0IvXo3LgVw1xW_IITyJ0g1_13RbxRY,49139
+nucleus/__init__.py,sha256=BCGkLpOSms6DhHHQLhm4F_Ne_1ZrbH84R2PHjtqBvgo,49405
 nucleus/annotation.py,sha256=qogOLOmNwv2o13kNEZkIRbI2lkp1owNZ2OnRsFJUZwU,42904
 nucleus/annotation_uploader.py,sha256=ipXw7QhJQXqaftHrbJn8oCGcm6aXyoL0GhQA9EcZWD4,9580
 nucleus/async_job.py,sha256=yjPDwyyLIrF0K67anGB40xux1AMhWrq1X_hPvQ_ewzc,6890
@@ -24,7 +24,7 @@ nucleus/data_transfer_object/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 nucleus/data_transfer_object/dataset_details.py,sha256=1YGvfKkPSqDrXK_y5mBXyRThY07tU-nwOCYTkYCSl6k,214
 nucleus/data_transfer_object/dataset_info.py,sha256=5P_gpvAyaqXxj2ZQuzLkGN2XROaN9Me56OLybCmO3R4,940
 nucleus/data_transfer_object/dataset_size.py,sha256=oe-dXaMLpsQRDcJQRZ9Ja8JTagYz4dviZuTognEylp0,111
-nucleus/data_transfer_object/job_status.py,sha256=9iXssqVwE0psfVPY3jxz56mjo0YZNxbvslSXqejSnlI,1830
+nucleus/data_transfer_object/job_status.py,sha256=hxvyNdrdVdj3UpEfwvryKC_QCJQEC9ru6IPjhPFcK44,2038
 nucleus/data_transfer_object/scenes_list.py,sha256=iTHE6vA47bRB6ciyEU4LArUXEXco4ArnGvZTGTeK8xs,432
 nucleus/dataset.py,sha256=6eGiXcsBt647dS_VfAH4rBNEYM6W3BODEnDkur43ZcM,86937
 nucleus/dataset_item.py,sha256=lm09hczNuGdqVx_iz1bFqD6IEni_cAosAyiSkmDs-eE,10077
@@ -57,23 +57,23 @@ nucleus/model_run.py,sha256=WtGy8cD86M_6aVNAp0ELZgunwbztNeOO8nv8ZSpbwhY,9280
 nucleus/package_not_installed.py,sha256=1ae0aqKAM3KrB0C-5MuPPXoz9tLWJUKtP1UZ-vw9Zik,1117
 nucleus/payload_constructor.py,sha256=ukoL4ivVGHr4hVuLXqMFcW_zR6JPV8VwlMT-pED8euM,4719
 nucleus/prediction.py,sha256=vLI0_ExaazVTrVhFru4mIWS_rX1xRFkg_El-5EAoaOQ,31092
-nucleus/pydantic_base.py,sha256=HgemyQIc-RfEVxGkTdApplZViAVMHBwk2G2SQzn7cMM,1151
+nucleus/pydantic_base.py,sha256=ZBUVrf948qzaxSuTaiDWxPC_Y8AOBdLKfi52ozGpGWk,1388
 nucleus/quaternion.py,sha256=TAnwj4arQXoTeofFgZMdZsCyxAMnu23N6to0F1WFNwk,1111
 nucleus/retry_strategy.py,sha256=daKZqjZYCh87WtXoVUuR9BZu2TTE-CtOFEYZ-d6xVMY,312
 nucleus/scene.py,sha256=qZQD7QdF6Ics8kuszsl278NCowKVnAkVNGHvPr5luRo,26937
 nucleus/slice.py,sha256=C7NRGuMmrJR6uv66xKD1MLaW_K2ocSwzp17JqWdIpZE,28005
-nucleus/test_launch_integration.py,sha256=O3-no-g05epYPvqMmjJiLLfITuxUq0wxHqhmFyUbEpo,10386
+nucleus/test_launch_integration.py,sha256=oFKLZWjFGeUvwVV0XAAjP1Y_oKFkaouh_SXVPXtCvcE,10688
 nucleus/track.py,sha256=ROmOyzYZKrHVTnLBhnk-qEBtklD_EDsSnRcGYE8xG4E,3247
 nucleus/upload_response.py,sha256=wR_pfZCBju1vGiGqbVgk8zhM6GhD3ebYxyGBm8y0GvY,3287
 nucleus/url_utils.py,sha256=EZ3vy1FYTGvXRVNyq43Wif-ypS8LFoDrYMYJre_DVuQ,790
-nucleus/utils.py,sha256=Ur_O4U0-dS0I0u6uw1pK6JaKZxZgpXkRz9AYJFNHqDE,15825
+nucleus/utils.py,sha256=A6kLL2THcu5UOwQY-B9K9eDVO3vDFkCwPlnpBX15WBY,15841
 nucleus/validate/__init__.py,sha256=UZx1tJHCRCosPXKdjFGaeifHOIf9R6ncYMcq77Gom54,786
 nucleus/validate/client.py,sha256=c8iF-fi7CEyKSoutzwJD0rhE6DwDDKFF-xC0wITCTjE,8219
 nucleus/validate/constants.py,sha256=EoR0BLKqQroyVx6TpirbcMKunOnREdtqL9OUvpejgAk,1089
 nucleus/validate/data_transfer_objects/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nucleus/validate/data_transfer_objects/eval_function.py,sha256=qNCibIw0Za7-er_bBO6ApSxXs96DhlKwzyJnYx7Tl64,4024
-nucleus/validate/data_transfer_objects/scenario_test.py,sha256=pJ2Wc1Wr6bifs5Ig_k7ZNGepQ6XOiPWAPakkUIoq890,738
-nucleus/validate/data_transfer_objects/scenario_test_evaluations.py,sha256=zyIjPvKeeQprenBYspi3zp8fyd9El8q6v9rypp9dB6A,1289
+nucleus/validate/data_transfer_objects/eval_function.py,sha256=3XSuw-7W3BwNR5u-wYunLj3peXLx3y-p_M4yK-UKNlM,4218
+nucleus/validate/data_transfer_objects/scenario_test.py,sha256=auDldQTSwzxLA1QF6dN2ghU_qMakQP64MEU5iIor0nw,932
+nucleus/validate/data_transfer_objects/scenario_test_evaluations.py,sha256=EQU3Gqirz2jwSHuIkdssn84tZ26q0NXehmis4K-18mM,1541
 nucleus/validate/data_transfer_objects/scenario_test_metric.py,sha256=jXGwPx0Y99ga0RfzwGMCG3_06NkiFsp0yHxUuGFJ2EI,208
 nucleus/validate/errors.py,sha256=UOI4oPeVzNLEmibWayfMdCMFNZvgsq-VRG42vC1cT_w,110
 nucleus/validate/eval_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -85,8 +85,8 @@ nucleus/validate/scenario_test.py,sha256=pCmM157dblSciZCDTw-f47Fpy3OUZFgXmokdhIL
 nucleus/validate/scenario_test_evaluation.py,sha256=Q0WzaEE9uUbPVc4EHlCoKjhJcqMNt4QbyiiJx12VOR0,4075
 nucleus/validate/scenario_test_metric.py,sha256=AhVFOB1ULwBqlZ2X_Au1TXy4iQELljtzR4ZpeLB35So,1209
 nucleus/validate/utils.py,sha256=VjdIJj9Pii4z4L6xbvClAc7ra_J7cX0vWB_J2X6yrGE,185
-scale_nucleus-0.16.13.dist-info/LICENSE,sha256=jaTGyQSQIZeWMo5iyYqgbAYHR9Bdy7nOzgE-Up3m_-g,1075
-scale_nucleus-0.16.13.dist-info/METADATA,sha256=FCqAu0aB9WuR9UHNoHcPuGqUxLinyA6IjFdF4HurHTs,7863
-scale_nucleus-0.16.13.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-scale_nucleus-0.16.13.dist-info/entry_points.txt,sha256=fmqEzh6NZQyg9eFMILnWabKT8OWQTMSCdDzMiVq2zYs,32
-scale_nucleus-0.16.13.dist-info/RECORD,,
+scale_nucleus-0.16.14.dist-info/LICENSE,sha256=jaTGyQSQIZeWMo5iyYqgbAYHR9Bdy7nOzgE-Up3m_-g,1075
+scale_nucleus-0.16.14.dist-info/METADATA,sha256=6roMP1Ge2m2Clwno_PZi6nN7zb2TZpoqe2zOCxlOaUo,7856
+scale_nucleus-0.16.14.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+scale_nucleus-0.16.14.dist-info/entry_points.txt,sha256=fmqEzh6NZQyg9eFMILnWabKT8OWQTMSCdDzMiVq2zYs,32
+scale_nucleus-0.16.14.dist-info/RECORD,,