Use chain.from_iterable

Author: Hanzhi Wang

pyiceberg/table/__init__.py

@@ -1998,7 +1998,7 @@ def plan_files(self) -> Iterable[FileScanTask]:
 
         residual_evaluators: Dict[int, Callable[[DataFile], ResidualEvaluator]] = KeyDefaultDict(self._build_residual_evaluator)
 
-        for manifest_entry in chain(*self.scan_plan_helper()):
+        for manifest_entry in chain.from_iterable(self.scan_plan_helper()):
             data_file = manifest_entry.data_file
             if data_file.content == DataFileContent.DATA:
                 data_entries.append(manifest_entry)

pyiceberg/table/inspect.py

@@ -310,7 +310,7 @@ def partitions(
 
         partitions_map: Dict[Tuple[str, Any], Any] = {}
 
-        for entry in itertools.chain(*scan.scan_plan_helper()):
+        for entry in itertools.chain.from_iterable(scan.scan_plan_helper()):
             partition = entry.data_file.partition
             partition_record_dict = {
                 field.name: partition[pos] for pos, field in enumerate(self.tbl.metadata.specs()[entry.data_file.spec_id].fields)