Typing of Dataset

xarray/backends/api.py

@@ -366,7 +366,7 @@ def _dataset_from_backend_dataset(
 
 
 def open_dataset(
-    filename_or_obj: str | os.PathLike,
+    filename_or_obj: str | os.PathLike | AbstractDataStore,
     *,
     engine: T_Engine = None,
     chunks: T_Chunks = None,

xarray/backends/cfgrib_.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import warnings
 

xarray/backends/file_manager.py

@@ -1,8 +1,10 @@
+from __future__ import annotations
+
 import contextlib
 import io
 import threading
 import warnings
-from typing import Any, Dict
+from typing import Any
 
 from ..core import utils
 from ..core.options import OPTIONS
@@ -16,7 +18,7 @@
 assert FILE_CACHE.maxsize, "file cache must be at least size one"
 
 
-REF_COUNTS: Dict[Any, int] = {}
+REF_COUNTS: dict[Any, int] = {}
 
 _DEFAULT_MODE = utils.ReprObject("<unused>")
 

xarray/backends/h5netcdf_.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 import io
 import os

xarray/backends/locks.py

@@ -1,7 +1,9 @@
+from __future__ import annotations
+
 import multiprocessing
 import threading
 import weakref
-from typing import Any, MutableMapping, Optional
+from typing import Any, MutableMapping
 
 try:
     from dask.utils import SerializableLock
@@ -62,7 +64,7 @@ def _get_lock_maker(scheduler=None):
     return _LOCK_MAKERS[scheduler]
 
 
-def _get_scheduler(get=None, collection=None) -> Optional[str]:
+def _get_scheduler(get=None, collection=None) -> str | None:
     """Determine the dask scheduler that is being used.
 
     None is returned if no dask scheduler is active.

xarray/backends/lru_cache.py

@@ -1,6 +1,8 @@
+from __future__ import annotations
+
 import threading
 from collections import OrderedDict
-from typing import Any, Callable, Iterator, MutableMapping, Optional, TypeVar
+from typing import Any, Callable, Iterator, MutableMapping, TypeVar
 
 K = TypeVar("K")
 V = TypeVar("V")
@@ -21,10 +23,10 @@ class LRUCache(MutableMapping[K, V]):
     the cache, e.g., ``cache.maxsize = new_size``.
     """
 
-    _cache: "OrderedDict[K, V]"
+    _cache: OrderedDict[K, V]
     _maxsize: int
     _lock: threading.RLock
-    _on_evict: Optional[Callable[[K, V], Any]]
+    _on_evict: Callable[[K, V], Any] | None
 
     __slots__ = ("_cache", "_lock", "_maxsize", "_on_evict")
 

xarray/backends/memory.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import copy
 
 import numpy as np

xarray/backends/netCDF4_.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 import operator
 import os

xarray/backends/netcdf3.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import unicodedata
 
 import numpy as np

xarray/backends/plugins.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 import inspect
 import itertools

xarray/backends/pseudonetcdf_.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 
 from ..core import indexing

xarray/backends/pydap_.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 
 from ..core import indexing

xarray/backends/pynio_.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 
 from ..core import indexing

xarray/backends/rasterio_.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import warnings
 

xarray/backends/scipy_.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import gzip
 import io
 import os

xarray/backends/store.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from .. import conventions
 from ..core.dataset import Dataset
 from .common import BACKEND_ENTRYPOINTS, AbstractDataStore, BackendEntrypoint

xarray/backends/zarr.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 import os
 import warnings

xarray/coding/calendar_ops.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 import pandas as pd
 

xarray/coding/frequencies.py

@@ -39,6 +39,7 @@
 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+from __future__ import annotations
 
 import numpy as np
 import pandas as pd

xarray/coding/strings.py

@@ -1,4 +1,6 @@
 """Coders for strings."""
+from __future__ import annotations
+
 from functools import partial
 
 import numpy as np

xarray/coding/times.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import re
 import warnings
 from datetime import datetime, timedelta
@@ -348,7 +350,7 @@ def _infer_time_units_from_diff(unique_timedeltas):
     return "seconds"
 
 
-def infer_calendar_name(dates) -> "CFCalendar":
+def infer_calendar_name(dates) -> CFCalendar:
     """Given an array of datetimes, infer the CF calendar name"""
     if is_np_datetime_like(dates.dtype):
         return "proleptic_gregorian"

xarray/coding/variables.py

@@ -1,4 +1,6 @@
 """Coders for individual Variable objects."""
+from __future__ import annotations
+
 import warnings
 from functools import partial
 from typing import Any, Hashable