line lengths

Author: hollymandel

xarray/core/dataarray.py

@@ -84,7 +84,10 @@
 )
 from xarray.plot.accessor import DataArrayPlotAccessor
 from xarray.plot.utils import _get_units_from_attrs
-from xarray.util.deprecation_helpers import _deprecate_positional_args, deprecate_dims
+from xarray.util.deprecation_helpers import (
+    _deprecate_positional_args,
+    deprecate_dims,
+)
 
 if TYPE_CHECKING:
     from dask.dataframe import DataFrame as DaskDataFrame
@@ -3925,7 +3928,9 @@ def to_pandas(self) -> Self | pd.Series | pd.DataFrame:
         return pandas_object
 
     def to_dataframe(
-        self, name: Hashable | None = None, dim_order: Sequence[Hashable] | None = None
+        self,
+        name: Hashable | None = None,
+        dim_order: Sequence[Hashable] | None = None,
     ) -> pd.DataFrame:
         """Convert this array and its coordinates into a tidy pandas.DataFrame.
 
@@ -4193,7 +4198,11 @@ def to_netcdf(
         --------
         Dataset.to_netcdf
         """
-        from xarray.backends.api import DATAARRAY_NAME, DATAARRAY_VARIABLE, to_netcdf
+        from xarray.backends.api import (
+            DATAARRAY_NAME,
+            DATAARRAY_VARIABLE,
+            to_netcdf,
+        )
 
         if self.name is None:
             # If no name is set then use a generic xarray name
@@ -4410,7 +4419,11 @@ def to_zarr(
         :ref:`io.zarr`
             The I/O user guide, with more details and examples.
         """
-        from xarray.backends.api import DATAARRAY_NAME, DATAARRAY_VARIABLE, to_zarr
+        from xarray.backends.api import (
+            DATAARRAY_NAME,
+            DATAARRAY_VARIABLE,
+            to_zarr,
+        )
 
         if self.name is None:
             # If no name is set then use a generic xarray name
@@ -4442,7 +4455,9 @@ def to_zarr(
         )
 
     def to_dict(
-        self, data: bool | Literal["list", "array"] = "list", encoding: bool = False
+        self,
+        data: bool | Literal["list", "array"] = "list",
+        encoding: bool = False,
     ) -> dict[str, Any]:
         """
         Convert this xarray.DataArray into a dictionary following xarray

xarray/core/dataset.py

@@ -38,7 +38,10 @@
 import pandas as pd
 
 from xarray.coding.calendar_ops import convert_calendar, interp_calendar
-from xarray.coding.cftimeindex import CFTimeIndex, _parse_array_of_cftime_strings
+from xarray.coding.cftimeindex import (
+    CFTimeIndex,
+    _parse_array_of_cftime_strings,
+)
 from xarray.core import (
     alignment,
     duck_array_ops,
@@ -126,10 +129,16 @@
     broadcast_variables,
     calculate_dimensions,
 )
-from xarray.namedarray.parallelcompat import get_chunked_array_type, guess_chunkmanager
+from xarray.namedarray.parallelcompat import (
+    get_chunked_array_type,
+    guess_chunkmanager,
+)
 from xarray.namedarray.pycompat import array_type, is_chunked_array
 from xarray.plot.accessor import DatasetPlotAccessor
-from xarray.util.deprecation_helpers import _deprecate_positional_args, deprecate_dims
+from xarray.util.deprecation_helpers import (
+    _deprecate_positional_args,
+    deprecate_dims,
+)
 
 if TYPE_CHECKING:
     from dask.dataframe import DataFrame as DaskDataFrame
@@ -140,7 +149,11 @@
     from xarray.backends.api import T_NetcdfEngine, T_NetcdfTypes
     from xarray.core.dataarray import DataArray
     from xarray.core.groupby import DatasetGroupBy
-    from xarray.core.merge import CoercibleMapping, CoercibleValue, _MergeResult
+    from xarray.core.merge import (
+        CoercibleMapping,
+        CoercibleValue,
+        _MergeResult,
+    )
     from xarray.core.resample import DatasetResample
     from xarray.core.rolling import DatasetCoarsen, DatasetRolling
     from xarray.core.types import (
@@ -253,7 +266,10 @@ def _get_chunk(var: Variable, chunks, chunkmanager: ChunkManagerEntrypoint):
     )
 
     chunk_shape = chunkmanager.normalize_chunks(
-        chunk_shape, shape=shape, dtype=var.dtype, previous_chunks=preferred_chunk_shape
+        chunk_shape,
+        shape=shape,
+        dtype=var.dtype,
+        previous_chunks=preferred_chunk_shape,
     )
 
     # Warn where requested chunks break preferred chunks, provided that the variable
@@ -887,7 +903,12 @@ def __dask_tokenize__(self) -> object:
         from dask.base import normalize_token
 
         return normalize_token(
-            (type(self), self._variables, self._coord_names, self._attrs or None)
+            (
+                type(self),
+                self._variables,
+                self._coord_names,
+                self._attrs or None,
+            )
         )
 
     def __dask_graph__(self):
@@ -1260,7 +1281,9 @@ def _overwrite_indexes(
             new_coord_names.remove(name)
 
         replaced = self._replace(
-            variables=new_variables, coord_names=new_coord_names, indexes=new_indexes
+            variables=new_variables,
+            coord_names=new_coord_names,
+            indexes=new_indexes,
         )
 
         if rename_dims:
@@ -2823,7 +2846,9 @@ def _resolve_frequency(
         return self._replace(variables)
 
     def _validate_indexers(
-        self, indexers: Mapping[Any, Any], missing_dims: ErrorOptionsWithWarn = "raise"
+        self,
+        indexers: Mapping[Any, Any],
+        missing_dims: ErrorOptionsWithWarn = "raise",
     ) -> Iterator[tuple[Hashable, int | slice | np.ndarray | Variable]]:
         """Here we make sure
         + indexer has a valid keys
@@ -4105,7 +4130,10 @@ def _validate_interp_indexer(x, new_x):
         # GH4739
         if obj.__dask_graph__():
             dask_indexers = {
-                k: (index.to_base_variable().chunk(), dest.to_base_variable().chunk())
+                k: (
+                    index.to_base_variable().chunk(),
+                    dest.to_base_variable().chunk(),
+                )
                 for k, (index, dest) in validated_indexers.items()
             }
 
@@ -4319,7 +4347,9 @@ def _rename_dims(self, name_dict: Mapping[Any, Hashable]) -> dict[Hashable, int]
         return {name_dict.get(k, k): v for k, v in self.sizes.items()}
 
     def _rename_indexes(
-        self, name_dict: Mapping[Any, Hashable], dims_dict: Mapping[Any, Hashable]
+        self,
+        name_dict: Mapping[Any, Hashable],
+        dims_dict: Mapping[Any, Hashable],
     ) -> tuple[dict[Hashable, Index], dict[Hashable, Variable]]:
         if not self._indexes:
             return {}, {}
@@ -4342,7 +4372,9 @@ def _rename_indexes(
         return indexes, variables
 
     def _rename_all(
-        self, name_dict: Mapping[Any, Hashable], dims_dict: Mapping[Any, Hashable]
+        self,
+        name_dict: Mapping[Any, Hashable],
+        dims_dict: Mapping[Any, Hashable],
     ) -> tuple[
         dict[Hashable, Variable],
         set[Hashable],
@@ -7389,7 +7421,8 @@ def _to_dataframe(self, ordered_dims: Mapping[Any, int]):
         ]
         index = self.coords.to_index([*ordered_dims])
         broadcasted_df = pd.DataFrame(
-            dict(zip(non_extension_array_columns, data, strict=True)), index=index
+            dict(zip(non_extension_array_columns, data, strict=True)),
+            index=index,
         )
         for extension_array_column in extension_array_columns:
             extension_array = self.variables[extension_array_column].data.array
@@ -7436,7 +7469,10 @@ def to_dataframe(self, dim_order: Sequence[Hashable] | None = None) -> pd.DataFr
         return self._to_dataframe(ordered_dims=ordered_dims)
 
     def _set_sparse_data_from_dataframe(
-        self, idx: pd.Index, arrays: list[tuple[Hashable, np.ndarray]], dims: tuple
+        self,
+        idx: pd.Index,
+        arrays: list[tuple[Hashable, np.ndarray]],
+        dims: tuple,
     ) -> None:
         from sparse import COO
 
@@ -7468,7 +7504,10 @@ def _set_sparse_data_from_dataframe(
             self[name] = (dims, data)
 
     def _set_numpy_data_from_dataframe(
-        self, idx: pd.Index, arrays: list[tuple[Hashable, np.ndarray]], dims: tuple
+        self,
+        idx: pd.Index,
+        arrays: list[tuple[Hashable, np.ndarray]],
+        dims: tuple,
     ) -> None:
         if not isinstance(idx, pd.MultiIndex):
             for name, values in arrays:
@@ -7591,7 +7630,9 @@ def from_dataframe(cls, dataframe: pd.DataFrame, sparse: bool = False) -> Self:
         return obj[dataframe.columns] if len(dataframe.columns) else obj
 
     def to_dask_dataframe(
-        self, dim_order: Sequence[Hashable] | None = None, set_index: bool = False
+        self,
+        dim_order: Sequence[Hashable] | None = None,
+        set_index: bool = False,
     ) -> DaskDataFrame:
         """
         Convert this dataset into a dask.dataframe.DataFrame.
@@ -7675,7 +7716,9 @@ def to_dask_dataframe(
         return df
 
     def to_dict(
-        self, data: bool | Literal["list", "array"] = "list", encoding: bool = False
+        self,
+        data: bool | Literal["list", "array"] = "list",
+        encoding: bool = False,
     ) -> dict[str, Any]:
         """
         Convert this dataset to a dictionary following xarray naming
@@ -7908,7 +7951,10 @@ def apply_over_both(lhs_data_vars, rhs_data_vars, lhs_vars, rhs_vars):
 
         if isinstance(other, Dataset):
             new_vars = apply_over_both(
-                self.data_vars, other.data_vars, self.variables, other.variables
+                self.data_vars,
+                other.data_vars,
+                self.variables,
+                other.variables,
             )
         else:
             other_variable = getattr(other, "variable", other)

xarray/core/missing.py

@@ -230,7 +230,10 @@ def _apply_over_vars_with_dim(func, self, dim=None, **kwargs):
 
 
 def get_clean_interp_index(
-    arr, dim: Hashable, use_coordinate: Hashable | bool = True, strict: bool = True
+    arr,
+    dim: Hashable,
+    use_coordinate: Hashable | bool = True,
+    strict: bool = True,
 ):
     """Return index to use for x values in interpolation or curve fitting.
 
@@ -343,7 +346,8 @@ def interp_na(
         if (
             dim in self._indexes
             and isinstance(
-                self._indexes[dim].to_pandas_index(), pd.DatetimeIndex | CFTimeIndex
+                self._indexes[dim].to_pandas_index(),
+                pd.DatetimeIndex | CFTimeIndex,
             )
             and use_coordinate
         ):
@@ -744,7 +748,8 @@ def interp_func(var, x, new_x, method: InterpOptions, kwargs):
         new_x0_shape = new_x[0].shape
         new_x0_chunks_is_not_none = new_x0_chunks is not None
         new_axes = {
-            ndim + i: new_x0_chunks[i] if new_x0_chunks_is_not_none else new_x0_shape[i]
+            ndim
+            + i: (new_x0_chunks[i] if new_x0_chunks_is_not_none else new_x0_shape[i])
             for i in range(new_x[0].ndim)
         }