scale-nucleus 0.16.13__tar.gz → 0.16.14__tar.gz

{scale_nucleus-0.16.13 → scale_nucleus-0.16.14}/PKG-INFO

@@ -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 → scale_nucleus-0.16.14}/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
         ]

{scale_nucleus-0.16.13 → scale_nucleus-0.16.14}/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

{scale_nucleus-0.16.13 → scale_nucleus-0.16.14}/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
 

{scale_nucleus-0.16.13 → scale_nucleus-0.16.14}/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(

{scale_nucleus-0.16.13 → scale_nucleus-0.16.14}/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}"
         )
 

{scale_nucleus-0.16.13 → scale_nucleus-0.16.14}/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}'")

{scale_nucleus-0.16.13 → scale_nucleus-0.16.14}/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}'")

{scale_nucleus-0.16.13 → scale_nucleus-0.16.14}/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 → scale_nucleus-0.16.14}/pyproject.toml

@@ -25,7 +25,7 @@ ignore = ["E501", "E741", "E731", "F401"]  # Easy ignore for getting it running
 
 [tool.poetry]
 name = "scale-nucleus"
-version = "0.16.13"
+version = "0.16.14"
 description = "The official Python client library for Nucleus, the Data Platform for AI"
 license =  "MIT"
 authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]
@@ -41,7 +41,7 @@ requests = "^2.23.0"
 tqdm = "^4.41.0"
 aiohttp = "^3.7.4"
 nest-asyncio = "^1.5.1"
-pydantic = "^1.8.2"
+pydantic = ">=1.8.2"
 numpy = [{ version = ">=1.19.5", python = ">=3.7,<3.10" }, { version = ">=1.22.0", python = ">=3.10"}]
 scipy = { version=">=1.4.1", optional = true }  # NOTE: COLAB has 1.4.1 and has problems updating
 click = ">=7.1.2,<9.0"  # NOTE: COLAB has 7.1.2 and has problems updating