use tbl.history() instead of ancestors_of()

We don't need to find all the ancestors, we only need to validate that the snapshot is an ancestor, i.e if it was ever current.

Author: chinmay-bhat

pyiceberg/table/__init__.py

@@ -113,7 +113,6 @@
     SnapshotLogEntry,
     SnapshotSummaryCollector,
     Summary,
-    ancestors_of,
     update_snapshot_summaries,
 )
 from pyiceberg.table.sorting import UNSORTED_SORT_ORDER, SortOrder
@@ -2020,10 +2019,7 @@ def rollback_to_snapshot(self, snapshot_id: int) -> ManageSnapshots:
         self._commit_if_ref_updates_exist()
         if self._transaction._table.snapshot_by_id(snapshot_id) is None:
             raise ValidationError(f"Cannot roll back to unknown snapshot id: {snapshot_id}")
-        if snapshot_id not in {
-            ancestor.snapshot_id
-            for ancestor in ancestors_of(self._transaction._table.current_snapshot(), self._transaction.table_metadata)
-        }:
+        if snapshot_id not in {ancestor.snapshot_id for ancestor in self._transaction._table.history()}:
             raise ValidationError(f"Cannot roll back to snapshot, not an ancestor of the current state: {snapshot_id}")
 
         update, requirement = self._transaction._set_ref_snapshot(snapshot_id=snapshot_id, ref_name="main", type="branch")