small fix

Author: chinmay-bhat

pyiceberg/table/snapshots.py

@@ -425,7 +425,7 @@ def ancestor_right_before_timestamp(
     current_snapshot: Optional[Snapshot], table_metadata: TableMetadata, timestamp_ms: int
 ) -> Optional[Snapshot]:
     """Get the ancestor right before the given timestamp."""
-    if current_snapshot is not None:
+    if current_snapshot:
         for ancestor in ancestors_of(current_snapshot, table_metadata):
             if ancestor.timestamp_ms < timestamp_ms:
                 return ancestor

tests/integration/test_snapshot_operations.py

@@ -53,7 +53,7 @@ def test_manage_snapshots_context_manager(catalog: Catalog) -> None:
     with tbl.manage_snapshots() as ms:
         ms.create_tag(snapshot_id=current_snapshot_id, tag_name="testing")
         ms.set_current_snapshot(snapshot_id=expected_snapshot_id)
-        ms.create_tag(snapshot_id=expected_snapshot_id, tag_name="testing2")
+        ms.create_branch(snapshot_id=expected_snapshot_id, branch_name="testing2")
     assert tbl.current_snapshot().snapshot_id is not current_snapshot_id  # type: ignore
     assert tbl.metadata.refs["testing"].snapshot_id == current_snapshot_id
     assert tbl.metadata.refs["main"] == SnapshotRef(snapshot_id=expected_snapshot_id, snapshot_ref_type="branch")