fix formatting

Author: trade6-bot

python/datafusion/context.py

@@ -45,7 +45,6 @@
 
     import pandas as pd
     import polars as pl
-    import pyarrow as pa
 
     from datafusion.plan import ExecutionPlan, LogicalPlan
 
@@ -1169,17 +1168,23 @@ def _convert_table_partition_cols(
                 elif data_type == "int":
                     converted_data_type = pa.int32()
                 else:
-                    raise ValueError(
-                        f"Unsupported literal data type '{data_type}' for partition column. Supported types are 'string' and 'int'"
+                    message = (
+                        f"Unsupported literal data type '{data_type}' for partition "
+                        "column. Supported types are 'string' and 'int'"
                     )
+                    raise ValueError(message)
             else:
                 converted_data_type = data_type
 
             converted_table_partition_cols.append((col, converted_data_type))
 
         if warn:
+            message = (
+                "using literals for table_partition_cols data types is deprecated,"
+                "use pyarrow types instead"
+            )
             warnings.warn(
-                "using literals for table_partition_cols data types is deprecated, use pyarrow types instead",
+                message,
                 category=DeprecationWarning,
                 stacklevel=2,
             )

python/tests/test_sql.py

@@ -157,7 +157,9 @@ def test_register_parquet(ctx, tmp_path):
     assert result.to_pydict() == {"cnt": [100]}
 
 
-@pytest.mark.parametrize("path_to_str,legacy_data_type", [(True, False), (False, False), (False, True)] )
+@pytest.mark.parametrize(
+    ("path_to_str", "legacy_data_type"), [(True, False), (False, False), (False, True)]
+)
 def test_register_parquet_partitioned(ctx, tmp_path, path_to_str, legacy_data_type):
     dir_root = tmp_path / "dataset_parquet_partitioned"
     dir_root.mkdir(exist_ok=False)
@@ -177,7 +179,7 @@ def test_register_parquet_partitioned(ctx, tmp_path, path_to_str, legacy_data_ty
 
     dir_root = str(dir_root) if path_to_str else dir_root
 
-    partition_data_type = 'string' if legacy_data_type else pa.string()
+    partition_data_type = "string" if legacy_data_type else pa.string()
 
     ctx.register_parquet(
         "datapp",