lint, simplify

Author: gabeiglio

pyiceberg/partitioning.py

@@ -225,8 +225,7 @@ def partition_type(self, schema: Schema) -> StructType:
         nested_fields = []
         schema_ids = index_by_id(schema)
         for field in self.fields:
-            if field.source_id in schema_ids:
-                source_field = schema.find_field(field.source_id)
+            if source_field := schema_ids.get(field.source_id):
                 result_type = field.transform.result_type(source_field.field_type)
                 nested_fields.append(NestedField(field.field_id, field.name, result_type, required=source_field.required))
             else:

tests/integration/test_reads.py

@@ -48,10 +48,8 @@
 from pyiceberg.io.pyarrow import (
     pyarrow_to_schema,
 )
-from pyiceberg.partitioning import PartitionField, PartitionSpec
 from pyiceberg.schema import Schema
 from pyiceberg.table import Table
-from pyiceberg.transforms import IdentityTransform
 from pyiceberg.types import (
     BinaryType,
     BooleanType,