Make bbox_size always single int

Author: carderne

src/satextractor/scheduler/scheduler.py

@@ -160,7 +160,7 @@ def cluster_tiles_in_utm(tiles: List[Tile], split_m: int) -> gpd.GeoDataFrame:
     splits = split_region_in_utm_tiles(
         tiles_geom_multi,
         crs=CRS.WGS84,
-        bbox_size=(split_m, split_m),
+        bbox_size=split_m,
     )
 
     tiles_gdf = gpd.GeoDataFrame({"geometry": tiles_geom})

src/satextractor/tiler/tiler.py

@@ -1,5 +1,4 @@
 from typing import List
-from typing import Tuple
 from typing import Union
 
 import shapely
@@ -11,15 +10,15 @@
 def split_region_in_utm_tiles(
     region: Union[shapely.geometry.Polygon, shapely.geometry.MultiPolygon],
     crs: CRS = CRS.WGS84,
-    bbox_size: Tuple[int, int] = (10000, 10000),
+    bbox_size: int = 10000,
     **kwargs,
 ) -> List[Tile]:
     """Split a given geometry in squares measured in meters.
     It splits the region in utm grid and the convert back to given crs.
 
     Args:
         region (UnionList[shapely.geometry.Polygon, shapely.geometry.MultiPolygon]): The region to split from
-        bbox_size (Tuple[int, int]): square bbox in meters
+        bbox_size (int): bbox size in meters
 
     Returns:
         [List[Tile]]: The Tiles representing each of the boxes
@@ -28,20 +27,15 @@ def split_region_in_utm_tiles(
     crs_bboxes = utm_splitter.get_bbox_list()
     info_bboxes = utm_splitter.get_info_list()
 
-    assert (
-        bbox_size[0] == bbox_size[1]
-    ), "bbox_size sides should be equal size, i.e. square"
-    size = bbox_size[0]
-
     tiles = []
     for info, box in zip(info_bboxes, crs_bboxes):
         # tile ids are globally unique and take the format shown below
         zone = info["utm_zone"]
         row = info["utm_row"]
-        x, y = (int(v / size) for v in box.lower_left)
+        x, y = (int(v / bbox_size) for v in box.lower_left)
         tiles.append(
             Tile(
-                id=f"{zone}_{row}_{size}_{x}_{y}",
+                id=f"{zone}_{row}_{bbox_size}_{x}_{y}",
                 epsg=box.crs.epsg,
                 bbox=(box.min_x, box.min_y, box.max_x, box.max_y),
             ),