Cleanup

Author: Fokko

pyiceberg/manifest.py

@@ -705,6 +705,7 @@ def fetch_manifest_entry(self, io: FileIO, discard_deleted: bool = True) -> List
             An Iterator of manifest entries.
         """
         from pyiceberg_core import manifest
+
         bs = io.new_input(self.manifest_path).open().read()
         manifest = manifest.read_manifest_entries(bs)
 
@@ -747,7 +748,6 @@ def _convert_entry(entry: Any) -> ManifestEntry:
         ]
 
 
-
 @cached(cache=LRUCache(maxsize=128), key=lambda io, manifest_list: hashkey(manifest_list))
 def _manifests(io: FileIO, manifest_list: str) -> Tuple[ManifestFile, ...]:
     """Read and cache manifests from the given manifest list, returning a tuple to prevent modification."""

tests/conftest.py

@@ -24,7 +24,7 @@
 and the built-in pytest fixture request should be used as an additional argument in the function. The fixture can then be
 retrieved using `request.getfixturevalue(fixture_name)`.
 """
-import json
+
 import os
 import re
 import socket
@@ -61,7 +61,7 @@
 )
 from pyiceberg.io.fsspec import FsspecFileIO
 from pyiceberg.manifest import DataFile, FileFormat
-from pyiceberg.partitioning import UNPARTITIONED_PARTITION_SPEC, PartitionSpec, PartitionField
+from pyiceberg.partitioning import PartitionField, PartitionSpec
 from pyiceberg.schema import Accessor, Schema
 from pyiceberg.serializers import ToOutputFile
 from pyiceberg.table import FileScanTask, Table
@@ -1850,9 +1850,8 @@ def simple_map() -> MapType:
 
 @pytest.fixture(scope="session")
 def test_schema() -> Schema:
-    return Schema(
-        NestedField(1, "VendorID", IntegerType(), False), NestedField(2, "tpep_pickup_datetime", IntegerType(), False)
-    )
+    return Schema(NestedField(1, "VendorID", IntegerType(), False), NestedField(2, "tpep_pickup_datetime", IntegerType(), False))
+
 
 @pytest.fixture(scope="session")
 def test_partition_spec() -> Schema:
@@ -1861,16 +1860,27 @@ def test_partition_spec() -> Schema:
         PartitionField(2, 1001, IdentityTransform(), "tpep_pickup_datetime"),
     )
 
+
 @pytest.fixture(scope="session")
-def generated_manifest_entry_file(avro_schema_manifest_entry: Dict[str, Any], test_schema: Schema, test_partition_spec: PartitionSpec) -> Generator[str, None, None]:
+def generated_manifest_entry_file(
+    avro_schema_manifest_entry: Dict[str, Any], test_schema: Schema, test_partition_spec: PartitionSpec
+) -> Generator[str, None, None]:
     from fastavro import parse_schema, writer
 
     parsed_schema = parse_schema(avro_schema_manifest_entry)
 
     with TemporaryDirectory() as tmpdir:
         tmp_avro_file = tmpdir + "/manifest.avro"
         with open(tmp_avro_file, "wb") as out:
-            writer(out, parsed_schema, manifest_entry_records, metadata={'schema': test_schema.model_dump_json(), 'partition-spec': test_partition_spec.fields,})
+            writer(
+                out,
+                parsed_schema,
+                manifest_entry_records,
+                metadata={
+                    "schema": test_schema.model_dump_json(),
+                    "partition-spec": test_partition_spec.fields,
+                },
+            )
         yield tmp_avro_file