style: sort imports and wrap long line after substrait.dataframe move

ruff check --fix / ruff format: substrait.dataframe.* imports now sort
ahead of substrait.extension_registry, and the reclaimed-name assert
message exceeded the line length.

Author: nielspardon

src/substrait/dataframe/__init__.py

@@ -83,7 +83,6 @@
     unique,
     when,
 )
-from substrait.extension_registry import ExtensionRegistry
 from substrait.dataframe.extension_relations import (
     ExtensionLeafDetail,
     ExtensionMultiDetail,
@@ -107,6 +106,7 @@
     update_table,
 )
 from substrait.dataframe.functions import f, functions_for
+from substrait.extension_registry import ExtensionRegistry
 
 __all__ = [
     # entry points

tests/dataframe/test_dtypes.py

@@ -98,7 +98,9 @@ def test_every_concrete_type_is_reachable_on_api():
         name = _KIND_TO_API.get(kind, kind)
         if name not in sub.__all__:
             missing.append(kind)
-    assert missing == [], f"Substrait types not exposed on substrait.dataframe: {missing}"
+    assert missing == [], (
+        f"Substrait types not exposed on substrait.dataframe: {missing}"
+    )
 
 
 def test_no_arg_types_are_datatypes_with_nullability():

tests/dataframe/test_frame.py

@@ -28,8 +28,8 @@
 from substrait.builders.plan import virtual_table as b_virtual_table
 from substrait.builders.plan import write_named_table as b_write
 from substrait.builders.type import fp64, i64, named_struct, string, struct
-from substrait.extension_registry import ExtensionRegistry
 from substrait.dataframe.frame import _JOIN_TYPES
+from substrait.extension_registry import ExtensionRegistry
 
 registry = ExtensionRegistry(load_default_extensions=True)
 

tests/dataframe/test_functions.py

@@ -8,8 +8,8 @@
 from substrait.builders.plan import consistent_partition_window
 from substrait.builders.plan import read_named_table as b_read
 from substrait.builders.type import fp64, i64, named_struct, string, struct
-from substrait.extension_registry import ExtensionRegistry
 from substrait.dataframe.functions import _safe_name
+from substrait.extension_registry import ExtensionRegistry
 
 registry = ExtensionRegistry(load_default_extensions=True)