icechunk 0.2.1__tar.gz → 0.2.2__tar.gz

{icechunk-0.2.1 → icechunk-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: icechunk
-Version: 0.2.1
+Version: 0.2.2
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "icechunk"
-version = "0.2.1"
+version = "0.2.2"
 description = "Transactional storage engine for Zarr designed for use on cloud object storage"
 readme = "README.md"
 repository = "https://github.com/earth-mover/icechunk"

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk/src/repository.rs

@@ -5,7 +5,9 @@ use std::{
     sync::Arc,
 };
 
+use async_recursion::async_recursion;
 use bytes::Bytes;
+use chrono::{DateTime, Utc};
 use err_into::ErrorInto as _;
 use futures::{
     stream::{FuturesOrdered, FuturesUnordered},
@@ -37,15 +39,13 @@ use crate::{
     Storage, StorageError,
 };
 
-#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
+#[derive(Debug, Clone, PartialEq, Eq)]
 #[non_exhaustive]
 pub enum VersionInfo {
-    #[serde(rename = "snapshot_id")]
     SnapshotId(SnapshotId),
-    #[serde(rename = "tag")]
     TagRef(String),
-    #[serde(rename = "branch")]
     BranchTipRef(String),
+    AsOf { branch: String, at: DateTime<Utc> },
 }
 
 #[derive(Debug, Error)]
@@ -60,6 +60,8 @@ pub enum RepositoryErrorKind {
 
     #[error("snapshot not found: `{id}`")]
     SnapshotNotFound { id: SnapshotId },
+    #[error("branch {branch} does not have a snapshots before or at {at}")]
+    InvalidAsOfSpec { branch: String, at: DateTime<Utc> },
     #[error("invalid snapshot id: `{0}`")]
     InvalidSnapshotId(String),
     #[error("tag error: `{0}`")]
@@ -404,11 +406,12 @@ impl Repository {
     }
 
     /// Returns the sequence of parents of the snapshot pointed by the given version
+    #[async_recursion(?Send)]
     #[instrument(skip(self))]
-    pub async fn ancestry(
-        &self,
+    pub async fn ancestry<'a>(
+        &'a self,
         version: &VersionInfo,
-    ) -> RepositoryResult<impl Stream<Item = RepositoryResult<SnapshotInfo>> + '_> {
+    ) -> RepositoryResult<impl Stream<Item = RepositoryResult<SnapshotInfo>> + 'a> {
         let snapshot_id = self.resolve_version(version).await?;
         self.snapshot_ancestry(&snapshot_id).await
     }
@@ -572,6 +575,24 @@ impl Repository {
                 .await?;
                 Ok(ref_data.snapshot)
             }
+            VersionInfo::AsOf { branch, at } => {
+                let tip = VersionInfo::BranchTipRef(branch.clone());
+                let snap = self
+                    .ancestry(&tip)
+                    .await?
+                    .try_skip_while(|parent| ready(Ok(&parent.flushed_at > at)))
+                    .take(1)
+                    .try_collect::<Vec<_>>()
+                    .await?;
+                match snap.into_iter().next() {
+                    Some(snap) => Ok(snap.id),
+                    None => Err(RepositoryErrorKind::InvalidAsOfSpec {
+                        branch: branch.clone(),
+                        at: *at,
+                    }
+                    .into()),
+                }
+            }
         }
     }
 

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk/src/storage/mod.rs

@@ -406,7 +406,11 @@ pub trait Storage: fmt::Debug + private::Sealed + Sync + Send {
     ) -> StorageResult<DateTime<Utc>>;
 
     async fn root_is_clean(&self) -> StorageResult<bool> {
-        Ok(self.list_objects(&Settings::default(), "").await?.next().await.is_none())
+        match self.list_objects(&Settings::default(), "").await?.next().await {
+            None => Ok(true),
+            Some(Ok(_)) => Ok(false),
+            Some(Err(err)) => Err(err),
+        }
     }
 
     async fn list_chunks(

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk/src/storage/s3.rs

@@ -205,9 +205,11 @@ impl S3Storage {
         let mut b =
             self.get_client().await.put_object().bucket(self.bucket.clone()).key(key);
 
-        if let Some(ct) = content_type {
-            b = b.content_type(ct)
-        };
+        if settings.unsafe_use_metadata() {
+            if let Some(ct) = content_type {
+                b = b.content_type(ct)
+            };
+        }
 
         if settings.unsafe_use_metadata() {
             for (k, v) in metadata {
@@ -312,9 +314,12 @@ impl Storage for S3Storage {
             .put_object()
             .bucket(self.bucket.clone())
             .key(key)
-            .content_type("application/yaml")
             .body(config.into());
 
+        if settings.unsafe_use_metadata() {
+            req = req.content_type("application/yaml")
+        }
+
         match (
             previous_version.etag(),
             settings.unsafe_use_conditional_create(),

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "icechunk-python"
-version = "0.2.1"
+version = "0.2.2"
 description = "Transactional storage engine for Zarr designed for use on cloud object storage"
 readme = "../README.md"
 repository = "https://github.com/earth-mover/icechunk"
@@ -21,7 +21,7 @@ crate-type = ["cdylib"]
 bytes = "1.9.0"
 chrono = { version = "0.4.39" }
 futures = "0.3.31"
-icechunk = { path = "../icechunk", version = "0.2.1", features = ["logs"] }
+icechunk = { path = "../icechunk", version = "0.2.2", features = ["logs"] }
 itertools = "0.14.0"
 pyo3 = { version = "0.23", features = [
   "chrono",

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/notebooks/demo-dummy-data.ipynb

@@ -335,7 +335,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 24,
+   "execution_count": null,
    "id": "d904f719-98cf-4f51-8e9a-1631dcb3fcba",
    "metadata": {},
    "outputs": [
@@ -348,7 +348,7 @@
     }
    ],
    "source": [
-    "session = repo.readonly_session(snapshot=first_commit)\n",
+    "session = repo.readonly_session(snapshot_id=first_commit)\n",
     "root_group = zarr.open_group(session.store, mode=\"r\")\n",
     "\n",
     "try:\n",

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/notebooks/version-control.ipynb

@@ -242,7 +242,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "id": "e785d9a1-36ec-4207-b334-20e0a68e3ac8",
    "metadata": {},
    "outputs": [
@@ -258,7 +258,7 @@
     }
    ],
    "source": [
-    "session = repo.readonly_session(snapshot=first_commit)\n",
+    "session = repo.readonly_session(snapshot_id=first_commit)\n",
     "root_group = zarr.open_group(store=session.store, mode=\"r\")\n",
     "dict(root_group.attrs)"
    ]

{icechunk-0.2.1 → icechunk-0.2.2/icechunk-python}/python/icechunk/__init__.py

@@ -14,6 +14,7 @@ from icechunk._icechunk_python import (
     ConflictType,
     Credentials,
     Diff,
+    GcsBearerCredential,
     GcsCredentials,
     GcsStaticCredentials,
     GCSummary,
@@ -95,6 +96,7 @@ __all__ = [
     "Credentials",
     "Diff",
     "GCSummary",
+    "GcsBearerCredential",
     "GcsCredentials",
     "GcsStaticCredentials",
     "IcechunkError",

{icechunk-0.2.1 → icechunk-0.2.2/icechunk-python}/python/icechunk/_icechunk_python.pyi

@@ -957,7 +957,7 @@ class PyRepository:
         *,
         branch: str | None = None,
         tag: str | None = None,
-        snapshot: str | None = None,
+        snapshot_id: str | None = None,
     ) -> AsyncIterator[SnapshotInfo]: ...
     def create_branch(self, branch: str, snapshot_id: str) -> None: ...
     def list_branches(self) -> set[str]: ...
@@ -972,17 +972,18 @@ class PyRepository:
         self,
         from_branch: str | None = None,
         from_tag: str | None = None,
-        from_snapshot: str | None = None,
+        from_snapshot_id: str | None = None,
         to_branch: str | None = None,
         to_tag: str | None = None,
-        to_snapshot: str | None = None,
+        to_snapshot_id: str | None = None,
     ) -> Diff: ...
     def readonly_session(
         self,
         branch: str | None = None,
         *,
         tag: str | None = None,
-        snapshot: str | None = None,
+        snapshot_id: str | None = None,
+        as_of: datetime.datetime | None = None,
     ) -> PySession: ...
     def writable_session(self, branch: str) -> PySession: ...
     def expire_snapshots(

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/python/icechunk/repository.py

@@ -196,7 +196,7 @@ class Repository:
         *,
         branch: str | None = None,
         tag: str | None = None,
-        snapshot: str | None = None,
+        snapshot_id: str | None = None,
     ) -> Iterator[SnapshotInfo]:
         """
         Get the ancestry of a snapshot.
@@ -207,7 +207,7 @@ class Repository:
             The branch to get the ancestry of.
         tag : str, optional
             The tag to get the ancestry of.
-        snapshot : str, optional
+        snapshot_id : str, optional
             The snapshot ID to get the ancestry of.
 
         Returns
@@ -223,7 +223,9 @@ class Repository:
         # the returned object is both an Async and Sync iterator
         res = cast(
             Iterator[SnapshotInfo],
-            self._repository.async_ancestry(branch=branch, tag=tag, snapshot=snapshot),
+            self._repository.async_ancestry(
+                branch=branch, tag=tag, snapshot_id=snapshot_id
+            ),
         )
         return res
 
@@ -232,7 +234,7 @@ class Repository:
         *,
         branch: str | None = None,
         tag: str | None = None,
-        snapshot: str | None = None,
+        snapshot_id: str | None = None,
     ) -> AsyncIterator[SnapshotInfo]:
         """
         Get the ancestry of a snapshot.
@@ -243,7 +245,7 @@ class Repository:
             The branch to get the ancestry of.
         tag : str, optional
             The tag to get the ancestry of.
-        snapshot : str, optional
+        snapshot_id : str, optional
             The snapshot ID to get the ancestry of.
 
         Returns
@@ -255,7 +257,9 @@ class Repository:
         -----
         Only one of the arguments can be specified.
         """
-        return self._repository.async_ancestry(branch=branch, tag=tag, snapshot=snapshot)
+        return self._repository.async_ancestry(
+            branch=branch, tag=tag, snapshot_id=snapshot_id
+        )
 
     def create_branch(self, branch: str, snapshot_id: str) -> None:
         """
@@ -336,7 +340,7 @@ class Repository:
         """
         self._repository.delete_branch(branch)
 
-    def delete_tag(self, branch: str) -> None:
+    def delete_tag(self, tag: str) -> None:
         """
         Delete a tag.
 
@@ -349,7 +353,7 @@ class Repository:
         -------
         None
         """
-        self._repository.delete_tag(branch)
+        self._repository.delete_tag(tag)
 
     def create_tag(self, tag: str, snapshot_id: str) -> None:
         """
@@ -400,10 +404,10 @@ class Repository:
         *,
         from_branch: str | None = None,
         from_tag: str | None = None,
-        from_snapshot: str | None = None,
+        from_snapshot_id: str | None = None,
         to_branch: str | None = None,
         to_tag: str | None = None,
-        to_snapshot: str | None = None,
+        to_snapshot_id: str | None = None,
     ) -> Diff:
         """
         Compute an overview of the operations executed from version `from` to version `to`.
@@ -421,10 +425,10 @@ class Repository:
         return self._repository.diff(
             from_branch=from_branch,
             from_tag=from_tag,
-            from_snapshot=from_snapshot,
+            from_snapshot_id=from_snapshot_id,
             to_branch=to_branch,
             to_tag=to_tag,
-            to_snapshot=to_snapshot,
+            to_snapshot_id=to_snapshot_id,
         )
 
     def readonly_session(
@@ -432,7 +436,8 @@ class Repository:
         branch: str | None = None,
         *,
         tag: str | None = None,
-        snapshot: str | None = None,
+        snapshot_id: str | None = None,
+        as_of: datetime.datetime | None = None,
     ) -> Session:
         """
         Create a read-only session.
@@ -447,8 +452,11 @@ class Repository:
             If provided, the branch to create the session on.
         tag : str, optional
             If provided, the tag to create the session on.
-        snapshot : str, optional
+        snapshot_id : str, optional
             If provided, the snapshot ID to create the session on.
+        as_of: datetime.datetime, optional
+            When combined with the branch argument, it will open the session at the last
+            snapshot that is at or before this datetime
 
         Returns
         -------
@@ -460,7 +468,9 @@ class Repository:
         Only one of the arguments can be specified.
         """
         return Session(
-            self._repository.readonly_session(branch=branch, tag=tag, snapshot=snapshot)
+            self._repository.readonly_session(
+                branch=branch, tag=tag, snapshot_id=snapshot_id, as_of=as_of
+            )
         )
 
     def writable_session(self, branch: str) -> Session:

{icechunk-0.2.1 → icechunk-0.2.2/icechunk-python}/python/icechunk/storage.py

@@ -2,6 +2,7 @@ from collections.abc import Callable
 from datetime import datetime
 
 from icechunk._icechunk_python import (
+    GcsBearerCredential,
     ObjectStoreConfig,
     S3Options,
     S3StaticCredentials,
@@ -222,6 +223,7 @@ def gcs_storage(
     bearer_token: str | None = None,
     from_env: bool | None = None,
     config: dict[str, str] | None = None,
+    get_credentials: Callable[[], GcsBearerCredential] | None = None,
 ) -> Storage:
     """Create a Storage instance that saves data in Google Cloud Storage object store.
 
@@ -235,6 +237,8 @@ def gcs_storage(
         Fetch credentials from the operative system environment
     bearer_token: str | None
         The bearer token to use for the object store
+    get_credentials: Callable[[], GcsBearerCredential] | None
+        Use this function to get and refresh object store credentials
     """
     credentials = gcs_credentials(
         service_account_file=service_account_file,
@@ -242,6 +246,7 @@ def gcs_storage(
         application_credentials=application_credentials,
         bearer_token=bearer_token,
         from_env=from_env,
+        get_credentials=get_credentials,
     )
     return Storage.new_gcs(
         bucket=bucket,

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/python/icechunk/store.py

@@ -289,7 +289,7 @@ class IcechunkStore(Store, SyncMixin):
 
         Parameters
         ----------
-        key : str
+        prefix : str
         """
         return await self._store.delete_dir(prefix)
 

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/src/repository.rs

@@ -506,18 +506,18 @@ impl PyRepository {
     }
 
     /// Returns an object that is both a sync and an async iterator
-    #[pyo3(signature = (*, branch = None, tag = None, snapshot = None))]
+    #[pyo3(signature = (*, branch = None, tag = None, snapshot_id = None))]
     pub fn async_ancestry(
         &self,
         py: Python<'_>,
         branch: Option<String>,
         tag: Option<String>,
-        snapshot: Option<String>,
+        snapshot_id: Option<String>,
     ) -> PyResult<PyAsyncGenerator> {
         let repo = Arc::clone(&self.0);
         // This function calls block_on, so we need to allow other thread python to make progress
         py.allow_threads(move || {
-            let version = args_to_version_info(branch, tag, snapshot)?;
+            let version = args_to_version_info(branch, tag, snapshot_id, None)?;
             let ancestry = pyo3_async_runtimes::tokio::get_runtime()
                 .block_on(async move { repo.ancestry_arc(&version).await })
                 .map_err(PyIcechunkStoreError::RepositoryError)?
@@ -689,20 +689,20 @@ impl PyRepository {
         })
     }
 
-    #[pyo3(signature = (*, from_branch=None, from_tag=None, from_snapshot=None, to_branch=None, to_tag=None, to_snapshot=None))]
+    #[pyo3(signature = (*, from_branch=None, from_tag=None, from_snapshot_id=None, to_branch=None, to_tag=None, to_snapshot_id=None))]
     #[allow(clippy::too_many_arguments)]
     pub fn diff(
         &self,
         py: Python<'_>,
         from_branch: Option<String>,
         from_tag: Option<String>,
-        from_snapshot: Option<String>,
+        from_snapshot_id: Option<String>,
         to_branch: Option<String>,
         to_tag: Option<String>,
-        to_snapshot: Option<String>,
+        to_snapshot_id: Option<String>,
     ) -> PyResult<PyDiff> {
-        let from = args_to_version_info(from_branch, from_tag, from_snapshot)?;
-        let to = args_to_version_info(to_branch, to_tag, to_snapshot)?;
+        let from = args_to_version_info(from_branch, from_tag, from_snapshot_id, None)?;
+        let to = args_to_version_info(to_branch, to_tag, to_snapshot_id, None)?;
 
         // This function calls block_on, so we need to allow other thread python to make progress
         py.allow_threads(move || {
@@ -717,17 +717,18 @@ impl PyRepository {
         })
     }
 
-    #[pyo3(signature = (*, branch = None, tag = None, snapshot = None))]
+    #[pyo3(signature = (*, branch = None, tag = None, snapshot_id = None, as_of = None))]
     pub fn readonly_session(
         &self,
         py: Python<'_>,
         branch: Option<String>,
         tag: Option<String>,
-        snapshot: Option<String>,
+        snapshot_id: Option<String>,
+        as_of: Option<DateTime<Utc>>,
     ) -> PyResult<PySession> {
         // This function calls block_on, so we need to allow other thread python to make progress
         py.allow_threads(move || {
-            let version = args_to_version_info(branch, tag, snapshot)?;
+            let version = args_to_version_info(branch, tag, snapshot_id, as_of)?;
             let session =
                 pyo3_async_runtimes::tokio::get_runtime().block_on(async move {
                     self.0
@@ -841,6 +842,7 @@ fn args_to_version_info(
     branch: Option<String>,
     tag: Option<String>,
     snapshot: Option<String>,
+    as_of: Option<DateTime<Utc>>,
 ) -> PyResult<VersionInfo> {
     let n = [&branch, &tag, &snapshot].iter().filter(|r| !r.is_none()).count();
     if n > 1 {
@@ -849,8 +851,18 @@ fn args_to_version_info(
         ));
     }
 
-    if let Some(branch_name) = branch {
-        Ok(VersionInfo::BranchTipRef(branch_name))
+    if as_of.is_some() && branch.is_none() {
+        return Err(PyValueError::new_err(
+            "as_of argument must be provided together with a branch name",
+        ));
+    }
+
+    if let Some(branch) = branch {
+        if let Some(at) = as_of {
+            Ok(VersionInfo::AsOf { branch, at })
+        } else {
+            Ok(VersionInfo::BranchTipRef(branch))
+        }
     } else if let Some(tag_name) = tag {
         Ok(VersionInfo::TagRef(tag_name))
     } else if let Some(snapshot_id) = snapshot {

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/tests/test_can_read_old.py

@@ -177,7 +177,7 @@ async def test_icechunk_can_read_old_repo() -> None:
         "Repository initialized",
     ]
     assert [
-        p.message for p in repo.ancestry(snapshot=main_snapshot)
+        p.message for p in repo.ancestry(snapshot_id=main_snapshot)
     ] == expected_main_history
 
     expected_branch_history = [
@@ -256,7 +256,7 @@ async def test_icechunk_can_read_old_repo() -> None:
     assert_array_equal(big_chunks[:], 42.0)
 
     parents = list(repo.ancestry(branch="main"))
-    diff = repo.diff(to_branch="main", from_snapshot=parents[-2].id)
+    diff = repo.diff(to_branch="main", from_snapshot_id=parents[-2].id)
     assert diff.new_groups == set()
     assert diff.new_arrays == set()
     assert set(diff.updated_chunks.keys()) == {

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/tests/test_credentials.py

@@ -113,7 +113,7 @@ class ExpirableCredentials:
         )
 
 
-def test_refreshable_credentials_refresh(tmp_path: Path) -> None:
+def test_s3_refreshable_credentials_refresh(tmp_path: Path) -> None:
     path = tmp_path / "calls.txt"
     creds_obj = ExpirableCredentials(path)
 

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/tests/test_stateful_repo_ops.py

@@ -271,7 +271,7 @@ class VersionControlStateMachine(RuleBasedStateMachine):
     @rule(ref=commits)
     def checkout_commit(self, ref):
         note(f"Checking out commit {ref}")
-        self.session = self.repo.readonly_session(snapshot=ref)
+        self.session = self.repo.readonly_session(snapshot_id=ref)
         assert self.session.read_only
         self.model.checkout_commit(ref)
 

{icechunk-0.2.1 → icechunk-0.2.2}/icechunk-python/tests/test_timetravel.py

@@ -58,14 +58,14 @@ def test_timetravel() -> None:
 
     new_snapshot_id = session.commit("commit 2")
 
-    session = repo.readonly_session(snapshot=first_snapshot_id)
+    session = repo.readonly_session(snapshot_id=first_snapshot_id)
     store = session.store
     group = zarr.open_group(store=store, mode="r")
     air_temp = cast(zarr.core.array.Array, group["air_temp"])
     assert store.read_only
     assert air_temp[200, 6] == 42
 
-    session = repo.readonly_session(snapshot=new_snapshot_id)
+    session = repo.readonly_session(snapshot_id=new_snapshot_id)
     store = session.store
     group = zarr.open_group(store=store, mode="r")
     air_temp = cast(zarr.core.array.Array, group["air_temp"])
@@ -116,7 +116,7 @@ def test_timetravel() -> None:
     air_temp = cast(zarr.core.array.Array, group["air_temp"])
     assert air_temp[200, 6] == 90
 
-    parents = list(repo.ancestry(snapshot=feature_snapshot_id))
+    parents = list(repo.ancestry(snapshot_id=feature_snapshot_id))
     assert [snap.message for snap in parents] == [
         "commit 3",
         "commit 2",
@@ -128,7 +128,7 @@ def test_timetravel() -> None:
     assert list(repo.ancestry(tag="v1.0")) == parents
     assert list(repo.ancestry(branch="feature-not-dead")) == parents
 
-    diff = repo.diff(to_tag="v1.0", from_snapshot=parents[-1].id)
+    diff = repo.diff(to_tag="v1.0", from_snapshot_id=parents[-1].id)
     assert diff.new_groups == {"/"}
     assert diff.new_arrays == {"/air_temp"}
     assert list(diff.updated_chunks.keys()) == ["/air_temp"]
@@ -185,11 +185,11 @@ Arrays deleted:
 
     with pytest.raises(ValueError, match="doesn't include"):
         # if we call diff in the wrong order it fails with a message
-        repo.diff(from_tag="v1.0", to_snapshot=parents[-1].id)
+        repo.diff(from_tag="v1.0", to_snapshot_id=parents[-1].id)
 
     # check async ancestry works
-    assert list(repo.ancestry(snapshot=feature_snapshot_id)) == asyncio.run(
-        async_ancestry(repo, snapshot=feature_snapshot_id)
+    assert list(repo.ancestry(snapshot_id=feature_snapshot_id)) == asyncio.run(
+        async_ancestry(repo, snapshot_id=feature_snapshot_id)
     )
     assert list(repo.ancestry(tag="v1.0")) == asyncio.run(
         async_ancestry(repo, tag="v1.0")
@@ -258,3 +258,38 @@ async def test_tag_delete() -> None:
 
     with pytest.raises(ValueError):
         repo.create_tag("tag", snap)
+
+
+async def test_session_with_as_of() -> None:
+    repo = ic.Repository.create(
+        storage=ic.in_memory_storage(),
+    )
+
+    session = repo.writable_session("main")
+    store = session.store
+
+    times = []
+    group = zarr.group(store=store, overwrite=True)
+    sid = session.commit("root")
+    times.append(next(repo.ancestry(snapshot_id=sid)).written_at)
+
+    for i in range(5):
+        session = repo.writable_session("main")
+        store = session.store
+        group = zarr.open_group(store=store)
+        group.create_group(f"child {i}")
+        sid = session.commit(f"child {i}")
+        times.append(next(repo.ancestry(snapshot_id=sid)).written_at)
+
+    ancestry = list(p for p in repo.ancestry(branch="main"))
+    assert len(ancestry) == 7  # initial + root + 5 children
+
+    store = repo.readonly_session("main", as_of=times[-1]).store
+    group = zarr.open_group(store=store, mode="r")
+
+    for i, time in enumerate(times):
+        store = repo.readonly_session("main", as_of=time).store
+        group = zarr.open_group(store=store, mode="r")
+        expected_children = {f"child {j}" for j in range(i)}
+        actual_children = {g[0] for g in group.members()}
+        assert expected_children == actual_children

{icechunk-0.2.1/icechunk-python → icechunk-0.2.2}/python/icechunk/__init__.py

@@ -14,6 +14,7 @@ from icechunk._icechunk_python import (
     ConflictType,
     Credentials,
     Diff,
+    GcsBearerCredential,
     GcsCredentials,
     GcsStaticCredentials,
     GCSummary,
@@ -95,6 +96,7 @@ __all__ = [
     "Credentials",
     "Diff",
     "GCSummary",
+    "GcsBearerCredential",
     "GcsCredentials",
     "GcsStaticCredentials",
     "IcechunkError",