fix pyright partially

Author: melonora

pixi.lock

@@ -76,10 +76,13 @@ tests = "pytest"
 doc_build = { cmd = "sphinx-apidoc -efT -o docs/_apidoc src/xarray_dataclasses && sphinx-build -a docs docs/_build" }
 
 [tool.pyright]
-reportImportCycles = "warning"
-reportUnknownArgumentType = "warning"
-reportUnknownMemberType = "warning"
-reportUnknownVariableType = "warning"
+reportMissingImports = "none"
+reportImportCycles = "none"
+reportUnknownArgumentType = "none"
+reportUnknownMemberType = "none"
+reportUnknownVariableType = "none"
+reportUnknownParameterType = "none"
+reportUntypedFunctionDecorator = "none"
 typeCheckingMode = "strict"
 
 [tool.pixi.pypi-dependencies]

pyproject.toml

@@ -8,6 +8,7 @@
 from typing import (
     Any,
     Dict,
+    get_type_hints,
     Hashable,
     List,
     Literal,
@@ -22,7 +23,7 @@
 # dependencies
 import numpy as np
 import xarray as xr
-from typing_extensions import ParamSpec, get_type_hints
+from typing_extensions import ParamSpec
 
 
 # submodules
@@ -133,7 +134,7 @@ def __post_init__(self) -> None:
         if model.names:
             setattr(self, "name", model.names[0].value)
 
-    def __call__(self, reference: Optional[AnyXarray] = None) -> xr.DataArray:
+    def __call__(self, reference: Optional[AnyXarray] = None) -> xr.DataArray:  # pyright: ignore[reportUnknownParameterType]
         """Create a DataArray object according to the entry."""
         from .dataarray import asdataarray
 
@@ -250,11 +251,11 @@ def get_entry(field: AnyField, value: Any) -> Optional[AnyEntry]:
             )
 
 
-def get_typedarray(
+def get_typedarray(  # pyright: ignore[reportUnknownParameterType]
     data: Any,
     dims: Dims,
-    dtype: Optional[AnyDType],
-    reference: Optional[AnyXarray] = None,
+    dtype: Optional[AnyDType],  # pyright: ignore[reportUnknownParameterType]
+    reference: Optional[AnyXarray] = None,  # pyright: ignore[reportUnknownParameterType]
 ) -> xr.DataArray:
     """Create a DataArray object with given dims and dtype.
 

src/xarray_dataclasses/datamodel.py

@@ -23,11 +23,15 @@
 from enum import Enum
 from itertools import chain
 from typing import (
+    Annotated,
     Any,
     ClassVar,
     Collection,
     Dict,
     Generic,
+    get_args,
+    get_origin,
+    get_type_hints,
     Hashable,
     Iterable,
     Literal,
@@ -44,14 +48,7 @@
 # dependencies
 import numpy as np
 import xarray as xr
-from typing_extensions import (
-    Annotated,
-    ParamSpec,
-    TypeAlias,
-    get_args,
-    get_origin,
-    get_type_hints,
-)
+from typing_extensions import ParamSpec, TypeAlias
 
 
 # type hints (private)
@@ -62,10 +59,10 @@
 TDType = TypeVar("TDType", covariant=True)
 THashable = TypeVar("THashable", bound=Hashable)
 
-AnyArray: TypeAlias = "np.ndarray[Any, Any]"
-AnyDType: TypeAlias = "np.dtype[Any]"
-AnyField: TypeAlias = "Field[Any]"
-AnyXarray: TypeAlias = "xr.DataArray | xr.Dataset"
+AnyArray: TypeAlias = np.ndarray
+AnyDType: TypeAlias = np.dtype
+AnyField: TypeAlias = Field[Any]
+AnyXarray: TypeAlias = Union[xr.DataArray, xr.Dataset]
 Dims = Tuple[str, ...]
 Order = Literal["C", "F"]
 Shape = Union[Sequence[int], int]
@@ -325,7 +322,7 @@ def get_dims(tp: Any) -> Dims:
     return tuple(str(get_args(arg)[0]) for arg in args)
 
 
-def get_dtype(tp: Any) -> Optional[AnyDType]:
+def get_dtype(tp: Any) -> Optional[AnyDType]:  # pyright: ignore[reportUnknownParameterType]
     """Extract a NumPy data type (dtype)."""
     try:
         dtype = get_args(get_args(get_annotated(tp))[1])[0]

src/xarray_dataclasses/typing.py

@@ -2,10 +2,10 @@
 from dataclasses import dataclass
 from typing import Literal, Tuple
 
-
 # dependencies
 import numpy as np
 import xarray as xr
+from typing_extensions import TypeAlias
 
 
 # submodules
@@ -24,11 +24,7 @@
 Y = Literal["y"]
 
 
-# dataclasses
-class Custom(xr.DataArray):
-    """Custom DataArray."""
-
-    __slots__ = ()
+Custom: TypeAlias = xr.DataArray
 
 
 @dataclass

tests/test_dataarray.py

@@ -6,6 +6,7 @@
 # dependencies
 import numpy as np
 import xarray as xr
+from typing_extensions import TypeAlias
 
 
 # submodules
@@ -24,9 +25,7 @@
 Y = Literal["y"]
 
 
-# dataclasses
-class Custom(xr.Dataset):
-    __slots__ = ()
+Custom: TypeAlias = xr.Dataset
 
 
 @dataclass