fix: apply ruff linter rules

tokoko · Aug 22, 2024 · d4cce20 · d4cce20
1 parent fbe1988
commit d4cce20
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ibis_substrait/tests/compiler/test_parity.py b/ibis_substrait/tests/compiler/test_parity.py
@@ -77,12 +77,11 @@ def run_query_duckdb_substrait(expr, datasets, compiler):
 
 
 def run_query_datafusion(expr, datasets, compiler):
-    import pyarrow
     import datafusion.substrait
 
     connection = datafusion.SessionContext()
 
-    for k, v in datasets.items():  # noqa: B007
+    for k, v in datasets.items():
         connection.deregister_table(k)
         connection.register_record_batches(k, [v.to_batches()])