format

Signed-off-by: Robert Kruszewski <github@robertk.io>

Author: robert3005

docs/conf.py

@@ -58,9 +58,7 @@
 nitpicky = True  # ensures all :class:, :obj:, etc. links are valid
 nitpick_ignore = []
 
-doctest_global_setup = (
-    "import pyarrow; import vortex; import vortex as vx; import random; random.seed(a=0)"
-)
+doctest_global_setup = "import pyarrow; import vortex; import vortex as vx; import random; random.seed(a=0)"
 doctest_default_flags = (
     doctest.ELLIPSIS | doctest.IGNORE_EXCEPTION_DETAIL | doctest.DONT_ACCEPT_TRUE_FOR_1 | doctest.NORMALIZE_WHITESPACE
 )

vortex-python/benchmark/test_aggregation.py

@@ -23,13 +23,7 @@ def test_arrow_table_aggregation(benchmark: BenchmarkFixture, vxf: vx.VortexFile
         for field in vxf.dtype.to_arrow_schema()  # pyright: ignore[reportUnknownVariableType]
         if _has_mean(field.type)  # pyright: ignore[reportUnknownMemberType, reportUnknownArgumentType]
     ]
-    benchmark(
-        lambda: (
-            pa.concat_tables(x.to_arrow_table() for x in vxf.scan())
-            .group_by([])
-            .aggregate(aggregations)
-        )
-    )
+    benchmark(lambda: pa.concat_tables(x.to_arrow_table() for x in vxf.scan()).group_by([]).aggregate(aggregations))
 
 
 @pytest.mark.benchmark(group="aggregation", disable_gc=True)

vortex-python/benchmark/test_filter.py

@@ -13,9 +13,7 @@
 
 @pytest.mark.benchmark(group="filter", disable_gc=True)
 def test_scan_filter(benchmark: BenchmarkFixture, vxf: vx.VortexFile):
-    benchmark(
-        lambda: pa.concat_tables(x.to_arrow_table() for x in vxf.scan(expr=column("x") >= 50_000))
-    )
+    benchmark(lambda: pa.concat_tables(x.to_arrow_table() for x in vxf.scan(expr=column("x") >= 50_000)))
 
 
 @pytest.mark.benchmark(group="filter", disable_gc=True)

vortex-python/benchmark/test_scalar_at.py

@@ -11,9 +11,7 @@
 
 @pytest.mark.benchmark(group="scalar_at", disable_gc=True)
 def test_scan_scalar_at(benchmark: BenchmarkFixture, vxf: vx.VortexFile):
-    benchmark(
-        lambda: pa.concat_tables(x.to_arrow_table() for x in vxf.scan(indices=vx.array([50_000])))
-    )
+    benchmark(lambda: pa.concat_tables(x.to_arrow_table() for x in vxf.scan(indices=vx.array([50_000]))))
 
 
 @pytest.mark.benchmark(group="scalar_at", disable_gc=True)

vortex-python/clickbench.py

@@ -632,9 +632,7 @@ def _iter():
                     arr = vx.Array.from_arrow(arr)
                     yield arr
 
-            it = vx.ArrayIterator.from_iter(
-                vx.DType.from_arrow(pf.schema_arrow, non_nullable=True), _iter()
-            )
+            it = vx.ArrayIterator.from_iter(vx.DType.from_arrow(pf.schema_arrow, non_nullable=True), _iter())
             vx.io.write(it, vx_path)
 
         lf = vx.open(vx_path).to_polars()