style(python): satisfy isort + black for PR lint gate

CI surfaced one isort failure (test_sample_bundle.py — `_bundle as
_bundle_mod` had to sort before the multiline `from … import (…)`
because `_` precedes letters in isort's ordering). Once that's fixed,
black also wants to reformat three pre-existing files where short
signatures had been unnecessarily split across multiple lines:

  src/databricks/labs/gbx/gridx/bng/functions.py    (3 fn signatures)
  src/databricks/labs/gbx/rasterx/functions.py     (1 fn signature)
  test/gridx/test_bng_functions.py                  (mechanical)

These are pure reformats — no behavior change. Verified locally with
the pinned versions from requirements-ci.in (isort==8.0.1, black==26.3.1,
flake8==7.3.0). All 26 .py files under src/ and test/ now pass all
three gates cleanly.

In test_sample_bundle.py, also replaced the previous pair of mis-
ordered comments above the imports with a single accurate comment;
isort's auto-fix had stacked both comments above the `_bundle_mod`
line, leaving "Public API from package" attached to the wrong import.

Co-authored-by: Isaac

Author: Michael Johns

python/geobrix/src/databricks/labs/gbx/gridx/bng/functions.py

@@ -160,9 +160,7 @@ def bng_euclideandistance(cell_id1: ColLike, cell_id2: ColLike) -> Column:
     Returns:
         Column of long (distance in metres).
     """
-    return f.call_function(
-        "gbx_bng_euclideandistance", _col(cell_id1), _col(cell_id2)
-    )
+    return f.call_function("gbx_bng_euclideandistance", _col(cell_id1), _col(cell_id2))
 
 
 def bng_geomkloop(geom: ColLike, resolution: ColLike, k: ColLike) -> Column:
@@ -176,9 +174,7 @@ def bng_geomkloop(geom: ColLike, resolution: ColLike, k: ColLike) -> Column:
     Returns:
         Column of array of BNG cell identifiers.
     """
-    return f.call_function(
-        "gbx_bng_geomkloop", _col(geom), _col(resolution), _col(k)
-    )
+    return f.call_function("gbx_bng_geomkloop", _col(geom), _col(resolution), _col(k))
 
 
 def bng_geomkring(geom: ColLike, resolution: ColLike, k: ColLike) -> Column:
@@ -192,9 +188,7 @@ def bng_geomkring(geom: ColLike, resolution: ColLike, k: ColLike) -> Column:
     Returns:
         Column of array of BNG cell identifiers.
     """
-    return f.call_function(
-        "gbx_bng_geomkring", _col(geom), _col(resolution), _col(k)
-    )
+    return f.call_function("gbx_bng_geomkring", _col(geom), _col(resolution), _col(k))
 
 
 def bng_kloop(cell_id: ColLike, k: ColLike) -> Column:

python/geobrix/src/databricks/labs/gbx/rasterx/functions.py

@@ -412,9 +412,7 @@ def rst_combineavg_agg(tile: ColLike) -> Column:
     return f.call_function("gbx_rst_combineavg_agg", _col(tile))
 
 
-def rst_derivedband_agg(
-    tile: ColLike, pyfunc: ColLike, func_name: ColLike
-) -> Column:
+def rst_derivedband_agg(tile: ColLike, pyfunc: ColLike, func_name: ColLike) -> Column:
     """Aggregate tiles and apply a Python UDF per band (use with groupBy).
 
     Args:
@@ -576,9 +574,7 @@ def rst_h3_rastertogridavg(tile: ColLike, resolution: ColLike) -> Column:
     Returns:
         Column of grid values (e.g. struct with H3 index and avg).
     """
-    return f.call_function(
-        "gbx_rst_h3_rastertogridavg", _col(tile), _col(resolution)
-    )
+    return f.call_function("gbx_rst_h3_rastertogridavg", _col(tile), _col(resolution))
 
 
 def rst_h3_rastertogridcount(tile: ColLike, resolution: ColLike) -> Column:
@@ -591,9 +587,7 @@ def rst_h3_rastertogridcount(tile: ColLike, resolution: ColLike) -> Column:
     Returns:
         Column of grid values (e.g. struct with H3 index and count).
     """
-    return f.call_function(
-        "gbx_rst_h3_rastertogridcount", _col(tile), _col(resolution)
-    )
+    return f.call_function("gbx_rst_h3_rastertogridcount", _col(tile), _col(resolution))
 
 
 def rst_h3_rastertogridmax(tile: ColLike, resolution: ColLike) -> Column:
@@ -606,9 +600,7 @@ def rst_h3_rastertogridmax(tile: ColLike, resolution: ColLike) -> Column:
     Returns:
         Column of grid values (e.g. struct with H3 index and max).
     """
-    return f.call_function(
-        "gbx_rst_h3_rastertogridmax", _col(tile), _col(resolution)
-    )
+    return f.call_function("gbx_rst_h3_rastertogridmax", _col(tile), _col(resolution))
 
 
 def rst_h3_rastertogridmin(tile: ColLike, resolution: ColLike) -> Column:
@@ -621,9 +613,7 @@ def rst_h3_rastertogridmin(tile: ColLike, resolution: ColLike) -> Column:
     Returns:
         Column of grid values (e.g. struct with H3 index and min).
     """
-    return f.call_function(
-        "gbx_rst_h3_rastertogridmin", _col(tile), _col(resolution)
-    )
+    return f.call_function("gbx_rst_h3_rastertogridmin", _col(tile), _col(resolution))
 
 
 def rst_h3_rastertogridmedian(tile: ColLike, resolution: ColLike) -> Column:
@@ -698,9 +688,7 @@ def rst_convolve(tile: ColLike, kernel: ColLike) -> Column:
     return f.call_function("gbx_rst_convolve", _col(tile), _col(kernel))
 
 
-def rst_derivedband(
-    tile_expr: ColLike, pyfunc: ColLike, func_name: ColLike
-) -> Column:
+def rst_derivedband(tile_expr: ColLike, pyfunc: ColLike, func_name: ColLike) -> Column:
     """Apply a Python UDF to each pixel (or band) to produce a derived band.
 
     Args:
@@ -792,14 +780,10 @@ def rst_ndvi(tile: ColLike, red_band: ColLike, nir_band: ColLike) -> Column:
     Returns:
         Column of raster tile (single-band NDVI).
     """
-    return f.call_function(
-        "gbx_rst_ndvi", _col(tile), _col(red_band), _col(nir_band)
-    )
+    return f.call_function("gbx_rst_ndvi", _col(tile), _col(red_band), _col(nir_band))
 
 
-def rst_rastertoworldcoord(
-    tile: ColLike, pixel_x: ColLike, pixel_y: ColLike
-) -> Column:
+def rst_rastertoworldcoord(tile: ColLike, pixel_x: ColLike, pixel_y: ColLike) -> Column:
     """Convert pixel (x, y) to world (x, y) in the CRS of the raster.
 
     Args:
@@ -889,9 +873,7 @@ def rst_updatetype(tile: ColLike, new_type: ColLike) -> Column:
     return f.call_function("gbx_rst_updatetype", _col(tile), _col(new_type))
 
 
-def rst_worldtorastercoord(
-    tile: ColLike, world_x: ColLike, world_y: ColLike
-) -> Column:
+def rst_worldtorastercoord(tile: ColLike, world_x: ColLike, world_y: ColLike) -> Column:
     """Convert world (x, y) to pixel (x, y) in the raster.
 
     Args:

python/geobrix/test/gridx/test_bng_functions.py

@@ -640,10 +640,12 @@ def test_bng_scalar_literal_args(spark, bng_registered):
     """
     polygon_wkt = "POLYGON ((530000 180000, 530500 180000, 530500 180500, 530000 180500, 530000 180000))"
     df = spark.range(1).select(
-        bng_registered.bng_pointascell(f.lit("POINT (400000 400000)"), 1).alias("cell_int_res"),
-        bng_registered.bng_pointascell(f.lit("POINT (400000 400000)"), f.lit("1km")).alias(
-            "cell_str_res"
+        bng_registered.bng_pointascell(f.lit("POINT (400000 400000)"), 1).alias(
+            "cell_int_res"
         ),
+        bng_registered.bng_pointascell(
+            f.lit("POINT (400000 400000)"), f.lit("1km")
+        ).alias("cell_str_res"),
         bng_registered.bng_polyfill(f.lit(polygon_wkt), 1).alias("cells_int_res"),
         bng_registered.bng_kloop(f.lit("TQ388791"), 1).alias("kloop_int_k"),
     )

python/geobrix/test/sample/test_sample_bundle.py

@@ -14,16 +14,14 @@
 
 import pytest
 
-# Public API from package
+# Public API + the _bundle module itself (needed for internal-helper coverage tests below).
+from databricks.labs.gbx.sample import _bundle as _bundle_mod
 from databricks.labs.gbx.sample import (
     get_temp_dir,
     get_volumes_path,
     run_complete_bundle,
     run_essential_bundle,
 )
-# Internal helpers for coverage
-from databricks.labs.gbx.sample import _bundle as _bundle_mod
-
 
 # ========== __init__ (package surface) ==========