Linting

Author: smaheshwar-pltr

pyiceberg/catalog/rest/__init__.py

@@ -974,7 +974,9 @@ def replace_table_transaction(
         )
 
         # Assign fresh schema IDs, reusing IDs from the existing schema by field name
-        fresh_schema, _ = assign_fresh_schema_ids_for_replace(iceberg_schema, existing_metadata.schema(), existing_metadata.last_column_id)
+        fresh_schema, _ = assign_fresh_schema_ids_for_replace(
+            iceberg_schema, existing_metadata.schema(), existing_metadata.last_column_id
+        )
 
         # Assign fresh partition spec IDs, reusing IDs from existing specs
         fresh_partition_spec, _ = assign_fresh_partition_spec_ids_for_replace(

tests/integration/test_rest_catalog.py

@@ -25,7 +25,6 @@
 from pyiceberg.exceptions import NoSuchTableError
 from pyiceberg.io.pyarrow import _dataframe_to_data_files
 from pyiceberg.schema import Schema
-from pyiceberg.table import StagedTable
 from pyiceberg.types import BooleanType, IntegerType, LongType, NestedField, StringType
 
 TEST_NAMESPACE_IDENTIFIER = "TEST NS"
@@ -104,8 +103,9 @@ def test_replace_table_transaction(catalog: Catalog, format_version: int) -> Non
                 snapshot_update.append_data_file(data_file)
 
     original.refresh()
-    assert original.current_snapshot() is not None
-    original_snapshot_id = original.current_snapshot().snapshot_id
+    current_snapshot = original.current_snapshot()
+    assert current_snapshot is not None
+    original_snapshot_id = current_snapshot.snapshot_id
 
     # Replace with a new schema
     new_schema = Schema(