Skip to content

Commit 056e6ea

Browse files
authored
Fix icechunk tests to opt-in to pickling and unpickling of Session and IcechunkStore following earth-mover/icechunk#652 (#688)
1 parent 3e5d6e2 commit 056e6ea

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

cubed/tests/test_icechunk.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -73,11 +73,13 @@ def test_store_icechunk(icechunk_storage, executor):
7373

7474
repo = Repository.create(storage=icechunk_storage)
7575
session = repo.writable_session("main")
76-
store = session.store
77-
78-
group = zarr.group(store=store, overwrite=True)
79-
target = group.create_array("a", shape=a.shape, dtype=a.dtype, chunks=a.chunksize)
80-
store_icechunk(session, sources=a, targets=target, executor=executor)
76+
with session.allow_pickling():
77+
store = session.store
78+
group = zarr.group(store=store, overwrite=True)
79+
target = group.create_array(
80+
"a", shape=a.shape, dtype=a.dtype, chunks=a.chunksize
81+
)
82+
store_icechunk(session, sources=a, targets=target, executor=executor)
8183
session.commit("commit 1")
8284

8385
# reopen store and check contents of array

0 commit comments

Comments
 (0)