Skip to content

Commit

Permalink
Refactor DaskManager's unify_chunks method***
Browse files Browse the repository at this point in the history
  • Loading branch information
andersy005 committed Feb 9, 2024
1 parent a7016ee commit be7109c
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions xarray/namedarray/daskmanager.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from __future__ import annotations

from collections.abc import Iterable, Sequence
from typing import TYPE_CHECKING, Any, Callable, cast
from typing import TYPE_CHECKING, Any, Callable

import numpy as np
from packaging.version import Version
Expand Down Expand Up @@ -236,10 +236,7 @@ def unify_chunks(
) -> tuple[dict[str, _NormalizedChunks], list[DaskArray]]:
from dask.array.core import unify_chunks

return cast(
tuple[dict[str, _NormalizedChunks], list[DaskArray]],
unify_chunks(*args, **kwargs), # type: ignore[no-untyped-call]
)
return (unify_chunks(*args, **kwargs),) # type: ignore[return-value, no-untyped-call]

def store(
self,
Expand Down

0 comments on commit be7109c

Please sign in to comment.