from metaflow import FlowSpec, step, current

class ZarrFlow(FlowSpec):

    @step
    def start(self):
        import zarr
        import numpy as np
        st = zarr.DirectoryStore('zarrdata/test')
        root = zarr.group(st, overwrite=True)
        ds = root.create_dataset('ds', shape=100, chunks=50, dtype='i8')
        ds[:] = np.arange(100)
        #zarr.convenience.consolidate_metadata(st)
        self.next(self.end)

    @step
    def end(self):
        pass

if __name__ == '__main__':
    ZarrFlow()