pre-commit run --all with ruff-format

asv_bench/benchmarks/__init__.py

@@ -67,7 +67,6 @@ def _skip_slow():
  >>> from . import _skip_slow
  >>> def time_something_slow():
  ... pass
- ...
  >>> time_something.setup = _skip_slow
  """
  if os.environ.get("ASV_SKIP_SLOW", "0") == "1":

doc/examples/apply_ufunc_vectorize_1d.ipynb

@@ -460,7 +460,8 @@
  "interped = interped.rename({\"new_lat\": \"lat\"})\n",
  "interped[\"lat\"] = newlat # need to add this manually\n",
  "xr.testing.assert_allclose(\n",
- " expected.transpose(*interped.dims), interped # order of dims is different\n",
+ " expected.transpose(*interped.dims),\n",
+ " interped, # order of dims is different\n",
  ")\n",
  "interped"
  ]

xarray/backends/api.py

@@ -1103,7 +1103,8 @@ def to_netcdf(
  *,
  multifile: Literal[True],
  invalid_netcdf: bool = False,
-) -> tuple[ArrayWriter, AbstractDataStore]: ...
+) -> tuple[ArrayWriter, AbstractDataStore]:
+ ...
 
 
 # path=None writes to bytes
@@ -1120,7 +1121,8 @@ def to_netcdf(
  compute: bool = True,
  multifile: Literal[False] = False,
  invalid_netcdf: bool = False,
-) -> bytes: ...
+) -> bytes:
+ ...
 
 
 # compute=False returns dask.Delayed
@@ -1138,7 +1140,8 @@ def to_netcdf(
  compute: Literal[False],
  multifile: Literal[False] = False,
  invalid_netcdf: bool = False,
-) -> Delayed: ...
+) -> Delayed:
+ ...
 
 
 # default return None
@@ -1155,7 +1158,8 @@ def to_netcdf(
  compute: Literal[True] = True,
  multifile: Literal[False] = False,
  invalid_netcdf: bool = False,
-) -> None: ...
+) -> None:
+ ...
 
 
 # if compute cannot be evaluated at type check time
@@ -1173,7 +1177,8 @@ def to_netcdf(
  compute: bool = False,
  multifile: Literal[False] = False,
  invalid_netcdf: bool = False,
-) -> Delayed | None: ...
+) -> Delayed | None:
+ ...
 
 
 # if multifile cannot be evaluated at type check time
@@ -1191,7 +1196,8 @@ def to_netcdf(
  compute: bool = False,
  multifile: bool = False,
  invalid_netcdf: bool = False,
-) -> tuple[ArrayWriter, AbstractDataStore] | Delayed | None: ...
+) -> tuple[ArrayWriter, AbstractDataStore] | Delayed | None:
+ ...
 
 
 # Any
@@ -1208,7 +1214,8 @@ def to_netcdf(
  compute: bool = False,
  multifile: bool = False,
  invalid_netcdf: bool = False,
-) -> tuple[ArrayWriter, AbstractDataStore] | bytes | Delayed | None: ...
+) -> tuple[ArrayWriter, AbstractDataStore] | bytes | Delayed | None:
+ ...
 
 
 def to_netcdf(
@@ -1642,7 +1649,8 @@ def to_zarr(
  zarr_version: int | None = None,
  write_empty_chunks: bool | None = None,
  chunkmanager_store_kwargs: dict[str, Any] | None = None,
-) -> backends.ZarrStore: ...
+) -> backends.ZarrStore:
+ ...
 
 
 # compute=False returns dask.Delayed
@@ -1665,7 +1673,8 @@ def to_zarr(
  zarr_version: int | None = None,
  write_empty_chunks: bool | None = None,
  chunkmanager_store_kwargs: dict[str, Any] | None = None,
-) -> Delayed: ...
+) -> Delayed:
+ ...
 
 
 def to_zarr(

xarray/backends/common.py

@@ -194,13 +194,10 @@ def load(self):
  For example::
 
  class SuffixAppendingDataStore(AbstractDataStore):
-
  def load(self):
  variables, attributes = AbstractDataStore.load(self)
- variables = {'%s_suffix' % k: v
- for k, v in variables.items()}
- attributes = {'%s_suffix' % k: v
- for k, v in attributes.items()}
+ variables = {"%s_suffix" % k: v for k, v in variables.items()}
+ attributes = {"%s_suffix" % k: v for k, v in attributes.items()}
  return variables, attributes
 
  This function will be called anytime variables or attributes

xarray/backends/locks.py

@@ -40,9 +40,9 @@ class SerializableLock:
  The creation of locks is itself not threadsafe.
  """
 
- _locks: ClassVar[WeakValueDictionary[Hashable, threading.Lock]] = (
- WeakValueDictionary()
- )
+ _locks: ClassVar[
+ WeakValueDictionary[Hashable, threading.Lock]
+ ] = WeakValueDictionary()
  token: Hashable
  lock: threading.Lock
 

xarray/backends/plugins.py

@@ -82,7 +82,7 @@ def backends_dict_from_pkg(
 
 
 def set_missing_parameters(
- backend_entrypoints: dict[str, type[BackendEntrypoint]]
+ backend_entrypoints: dict[str, type[BackendEntrypoint]],
 ) -> None:
  for _, backend in backend_entrypoints.items():
  if backend.open_dataset_parameters is None:
@@ -91,7 +91,7 @@ def set_missing_parameters(
 
 
 def sort_backends(
- backend_entrypoints: dict[str, type[BackendEntrypoint]]
+ backend_entrypoints: dict[str, type[BackendEntrypoint]],
 ) -> dict[str, type[BackendEntrypoint]]:
  ordered_backends_entrypoints = {}
  for be_name in STANDARD_BACKENDS_ORDER:

xarray/core/_typed_ops.py

@@ -455,163 +455,199 @@ def _binary_op(
  raise NotImplementedError
 
  @overload
- def __add__(self, other: T_DataArray) -> T_DataArray: ...
+ def __add__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __add__(self, other: VarCompatible) -> Self: ...
+ def __add__(self, other: VarCompatible) -> Self:
+ ...
 
  def __add__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.add)
 
  @overload
- def __sub__(self, other: T_DataArray) -> T_DataArray: ...
+ def __sub__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __sub__(self, other: VarCompatible) -> Self: ...
+ def __sub__(self, other: VarCompatible) -> Self:
+ ...
 
  def __sub__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.sub)
 
  @overload
- def __mul__(self, other: T_DataArray) -> T_DataArray: ...
+ def __mul__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __mul__(self, other: VarCompatible) -> Self: ...
+ def __mul__(self, other: VarCompatible) -> Self:
+ ...
 
  def __mul__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.mul)
 
  @overload
- def __pow__(self, other: T_DataArray) -> T_DataArray: ...
+ def __pow__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __pow__(self, other: VarCompatible) -> Self: ...
+ def __pow__(self, other: VarCompatible) -> Self:
+ ...
 
  def __pow__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.pow)
 
  @overload
- def __truediv__(self, other: T_DataArray) -> T_DataArray: ...
+ def __truediv__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __truediv__(self, other: VarCompatible) -> Self: ...
+ def __truediv__(self, other: VarCompatible) -> Self:
+ ...
 
  def __truediv__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.truediv)
 
  @overload
- def __floordiv__(self, other: T_DataArray) -> T_DataArray: ...
+ def __floordiv__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __floordiv__(self, other: VarCompatible) -> Self: ...
+ def __floordiv__(self, other: VarCompatible) -> Self:
+ ...
 
  def __floordiv__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.floordiv)
 
  @overload
- def __mod__(self, other: T_DataArray) -> T_DataArray: ...
+ def __mod__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __mod__(self, other: VarCompatible) -> Self: ...
+ def __mod__(self, other: VarCompatible) -> Self:
+ ...
 
  def __mod__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.mod)
 
  @overload
- def __and__(self, other: T_DataArray) -> T_DataArray: ...
+ def __and__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __and__(self, other: VarCompatible) -> Self: ...
+ def __and__(self, other: VarCompatible) -> Self:
+ ...
 
  def __and__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.and_)
 
  @overload
- def __xor__(self, other: T_DataArray) -> T_DataArray: ...
+ def __xor__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __xor__(self, other: VarCompatible) -> Self: ...
+ def __xor__(self, other: VarCompatible) -> Self:
+ ...
 
  def __xor__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.xor)
 
  @overload
- def __or__(self, other: T_DataArray) -> T_DataArray: ...
+ def __or__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __or__(self, other: VarCompatible) -> Self: ...
+ def __or__(self, other: VarCompatible) -> Self:
+ ...
 
  def __or__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.or_)
 
  @overload
- def __lshift__(self, other: T_DataArray) -> T_DataArray: ...
+ def __lshift__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __lshift__(self, other: VarCompatible) -> Self: ...
+ def __lshift__(self, other: VarCompatible) -> Self:
+ ...
 
  def __lshift__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.lshift)
 
  @overload
- def __rshift__(self, other: T_DataArray) -> T_DataArray: ...
+ def __rshift__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __rshift__(self, other: VarCompatible) -> Self: ...
+ def __rshift__(self, other: VarCompatible) -> Self:
+ ...
 
  def __rshift__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.rshift)
 
  @overload
- def __lt__(self, other: T_DataArray) -> T_DataArray: ...
+ def __lt__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __lt__(self, other: VarCompatible) -> Self: ...
+ def __lt__(self, other: VarCompatible) -> Self:
+ ...
 
  def __lt__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.lt)
 
  @overload
- def __le__(self, other: T_DataArray) -> T_DataArray: ...
+ def __le__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __le__(self, other: VarCompatible) -> Self: ...
+ def __le__(self, other: VarCompatible) -> Self:
+ ...
 
  def __le__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.le)
 
  @overload
- def __gt__(self, other: T_DataArray) -> T_DataArray: ...
+ def __gt__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __gt__(self, other: VarCompatible) -> Self: ...
+ def __gt__(self, other: VarCompatible) -> Self:
+ ...
 
  def __gt__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.gt)
 
  @overload
- def __ge__(self, other: T_DataArray) -> T_DataArray: ...
+ def __ge__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __ge__(self, other: VarCompatible) -> Self: ...
+ def __ge__(self, other: VarCompatible) -> Self:
+ ...
 
  def __ge__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, operator.ge)
 
  @overload # type:ignore[override]
- def __eq__(self, other: T_DataArray) -> T_DataArray: ...
+ def __eq__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __eq__(self, other: VarCompatible) -> Self: ...
+ def __eq__(self, other: VarCompatible) -> Self:
+ ...
 
  def __eq__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, nputils.array_eq)
 
  @overload # type:ignore[override]
- def __ne__(self, other: T_DataArray) -> T_DataArray: ...
+ def __ne__(self, other: T_DataArray) -> T_DataArray:
+ ...
 
  @overload
- def __ne__(self, other: VarCompatible) -> Self: ...
+ def __ne__(self, other: VarCompatible) -> Self:
+ ...
 
  def __ne__(self, other: VarCompatible) -> Self | T_DataArray:
  return self._binary_op(other, nputils.array_ne)