Skip to content

Commit

Permalink
removed unpacking for pydocstyle
Browse files Browse the repository at this point in the history
  • Loading branch information
Berkant03 committed Jan 27, 2025
1 parent 6eb55e9 commit bc93e18
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions heat/core/io.py
Original file line number Diff line number Diff line change
Expand Up @@ -1352,7 +1352,7 @@ def load_zarr(
offset, local_shape, slices = comm.chunk(shape, split)

return factories.array(
arr[*slices], dtype=dtype, is_split=split, device=device, comm=comm # noqa: E999
arr[slices], dtype=dtype, is_split=split, device=device, comm=comm
)

def save_zarr(path: str, dndarray: DNDarray, overwrite: bool = False, **kwargs) -> None:
Expand Down Expand Up @@ -1445,7 +1445,7 @@ def save_zarr(path: str, dndarray: DNDarray, overwrite: bool = False, **kwargs)
if dndarray.split is not None:
_, _, slices = MPI_WORLD.chunk(dndarray.gshape, dndarray.split)

zarr_array[*slices] = ( # noqa: E999
zarr_array[slices] = (
dndarray.larray.numpy() # Numpy array needed as zarr can only understand numpy dtypes and infers it.
)
else:
Expand Down

0 comments on commit bc93e18

Please sign in to comment.