style: pre-commit fixes

src/awkward/operations/ak_to_parquet.py

@@ -374,7 +374,6 @@ def parquet_columns(specifier, only=None):
     fs, destination = fsspec.core.url_to_fs(destination, **(storage_options or {}))
     metalist = []
 
-
     with pyarrow_parquet.ParquetWriter(
         destination,
         table.schema,
@@ -396,6 +395,7 @@ def parquet_columns(specifier, only=None):
     ) as writer:
         if iter:
             from pyarrow import Table
+
             iterator = batch_iterator(
                 layout,
                 list_to32,
@@ -440,7 +440,6 @@ def write_metadata(dir_path, fs, *metas, global_metadata=True):
             md.write_metadata_file(fil)
 
 
-
 def batch_iterator(
     layout,
     list_to32,
@@ -455,9 +454,7 @@ def batch_iterator(
 
     pyarrow = awkward._connect.pyarrow.import_pyarrow("ak.to_parquet")
 
-    if isinstance(
-        ak.operations.type(layout), ak.types.RecordType
-    ):
+    if isinstance(ak.operations.type(layout), ak.types.RecordType):
         names = layout.keys()
         contents = [layout[name] for name in names]
     else:
@@ -481,11 +478,7 @@ def batch_iterator(
         )
         pa_fields.append(
             pyarrow.field(name, pa_arrays[-1].type).with_nullable(
-                isinstance(
-                    ak.operations.type(content), ak.types.OptionType
-                )
+                isinstance(ak.operations.type(content), ak.types.OptionType)
             )
         )
-    yield pyarrow.RecordBatch.from_arrays(
-        pa_arrays, schema=pyarrow.schema(pa_fields)
-    )
+    yield pyarrow.RecordBatch.from_arrays(pa_arrays, schema=pyarrow.schema(pa_fields))

src/awkward/operations/ak_to_parquet_row_groups.py

@@ -1,14 +1,14 @@
-
 from __future__ import annotations
 
 from awkward import ak_to_parquet
 from awkward._dispatch import high_level_function
+
 __all__ = ("to_parquet",)
 
 
 @high_level_function()
 def to_parquet_row_groups(
-    array_iterator, # good or bad name?
+    array_iterator,  # good or bad name?
     destination,
     *,
     list_to32=False,
@@ -170,9 +170,9 @@ def to_parquet_row_groups(
     # Dispatch
     yield (array_iterator,)
 
-    # If the input is an iterator, then should row-group size still be 
-        # an option? Or should there be a check to determine if it's the same? 
-        # seems like it should be set based on the iterator or something
+    # If the input is an iterator, then should row-group size still be
+    # an option? Or should there be a check to determine if it's the same?
+    # seems like it should be set based on the iterator or something
 
     return ak_to_parquet._impl(
         array_iterator,
@@ -202,16 +202,21 @@ def to_parquet_row_groups(
         iter=True,
     )
 
-import awkward as ak
-from skhep_testdata import data_path
+
 import uproot
+from skhep_testdata import data_path
+
+import awkward as ak
+
 path = "/Users/zobil/Documents/awkward/tests/samples/array.parquet"
 
-iterator = uproot.iterate(uproot.open(data_path("uproot-HZZ.root"))['events'], step_size=10)
+iterator = uproot.iterate(
+    uproot.open(data_path("uproot-HZZ.root"))["events"], step_size=10
+)
 
 # ak.to_parquet(array, "/Users/zobil/Documents/awkward/tests/samples/array.parquet")
 to_parquet_row_groups(iterator, path, row_group_size=10)
 
 test = ak.from_parquet(path)
 
-print(len(test['Jet_Px']))
+print(len(test["Jet_Px"]))