Make the CI happy

Author: Fokko

pyiceberg/table/__init__.py

@@ -122,7 +122,6 @@
     _FastAppendFiles,
 )
 from pyiceberg.table.update.sorting import UpdateSortOrder
-from pyiceberg.table.update.snapshot import ManageSnapshots, UpdateSnapshot, _FastAppendFiles
 from pyiceberg.table.update.spec import UpdateSpec
 from pyiceberg.table.update.statistics import UpdateStatistics
 from pyiceberg.transforms import IdentityTransform
@@ -456,7 +455,9 @@ def update_sort_order(self, case_sensitive: bool = True) -> UpdateSortOrder:
             case_sensitive=case_sensitive,
         )
 
-    def update_snapshot(self, snapshot_properties: Dict[str, str] = EMPTY_DICT) -> UpdateSnapshot:
+    def update_snapshot(
+        self, snapshot_properties: Dict[str, str] = EMPTY_DICT, branch: Optional[str] = MAIN_BRANCH
+    ) -> UpdateSnapshot:
         """Create a new UpdateSnapshot to produce a new snapshot for the table.
 
         Returns: