chore: fmt

Author: JasperHG90

pyiceberg/table/update/sorting.py

@@ -16,9 +16,9 @@
 # under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, List, Tuple, Optional
+from typing import TYPE_CHECKING, Any, List, Optional, Tuple
 
-from pyiceberg.table.sorting import NullOrder, SortDirection, SortField, SortOrder, INITIAL_SORT_ORDER_ID, UNSORTED_SORT_ORDER
+from pyiceberg.table.sorting import INITIAL_SORT_ORDER_ID, UNSORTED_SORT_ORDER, NullOrder, SortDirection, SortField, SortOrder
 from pyiceberg.table.update import (
     AddSortOrderUpdate,
     AssertDefaultSortOrderId,
@@ -74,7 +74,7 @@ def _add_sort_field(
             )
         )
         return self
-    
+
     def _reuse_or_create_sort_order_id(self) -> int:
         """Return the last assigned sort order id or create a new one."""
         new_sort_order_id = INITIAL_SORT_ORDER_ID
@@ -123,7 +123,10 @@ def _commit(self) -> UpdatesAndRequirements:
         requirements: Tuple[TableRequirement, ...] = ()
         updates: Tuple[TableUpdate, ...] = ()
 
-        if self._transaction.table_metadata.default_sort_order_id != new_sort_order.order_id and self._transaction.table_metadata.sort_order_by_id(new_sort_order.order_id) is None:
+        if (
+            self._transaction.table_metadata.default_sort_order_id != new_sort_order.order_id
+            and self._transaction.table_metadata.sort_order_by_id(new_sort_order.order_id) is None
+        ):
             updates = (AddSortOrderUpdate(sort_order=new_sort_order), SetDefaultSortOrderUpdate(sort_order_id=-1))
         else:
             updates = (SetDefaultSortOrderUpdate(sort_order_id=new_sort_order.order_id),)

tests/integration/test_sort_order_update.py

@@ -130,10 +130,10 @@ def test_update_existing_sort_order(catalog: Catalog, format_version: str, table
         "bar", IdentityTransform(), NullOrder.NULLS_FIRST
     ).commit()
     # Go back to the first sort order
-    simple_table.update_sort_order().asc("foo", IdentityTransform(), NullOrder.NULLS_FIRST).commit()        
+    simple_table.update_sort_order().asc("foo", IdentityTransform(), NullOrder.NULLS_FIRST).commit()
     assert (
         len(simple_table.sort_orders()) == 3
-    ) # line 133 should not create a new sort order since it is the same as the first one
+    )  # line 133 should not create a new sort order since it is the same as the first one
     assert simple_table.sort_order() == SortOrder(
         SortField(source_id=1, transform=IdentityTransform(), direction=SortDirection.ASC, null_order=NullOrder.NULLS_FIRST),
         order_id=1,