Skip to content
Snippets Groups Projects
Commit 3cd4a84d authored by Siddhant Tibrewal's avatar Siddhant Tibrewal
Browse files

fixed chunk-shape in configure and added hiopy-worker dependencies in the init file

parent 4160449a
No related branches found
No related tags found
No related merge requests found
Pipeline #99543 waiting for manual action
from ._data_handler import *
from ._distribute_work import *
from ._grids import *
from .worker import * # noqa: F403
from .loco import * # noqa: F403
......@@ -86,7 +86,7 @@ def add_variable(
_shard_shape = None
if chunk_shape is not None:
_chunk_shape = (np.min(chunk_shape, shape),)
_chunk_shape = tuple(min(chunk_shape, shape))
if chunks_per_shard is not None:
_shard_shape = tuple(i * chunks_per_shard for i in _chunk_shape)
......
......@@ -8,10 +8,10 @@ import numpy as np
import zarr
from coyote import Coyote, group_comm_rank, group_comm_size, init, run, start_datetime
from ._data_handler import DataHandler
from ._distribute_work import distribute_work
from ._grids import def_grid, grid_id
from .loco import LocoServer
from _data_handler import DataHandler
from _distribute_work import distribute_work
from _grids import def_grid, grid_id
from loco import LocoServer
def main():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment