Merge pull request #2 from terrafloww/tile_async_improvements

Async improvements

examples/basic_workflow_xarray.py

@@ -14,7 +14,7 @@ def main():
     workspace_dir = Path.home() / "rasteret_workspace"
     workspace_dir.mkdir(exist_ok=True)
 
-    custom_name = "bangalore-v3"
+    custom_name = "bangalore"
     date_range = ("2024-03-01", "2024-03-31")
     data_source = DataSources.LANDSAT
 
@@ -23,14 +23,8 @@ def main():
         [(77.55, 13.01), (77.58, 13.01), (77.58, 13.08), (77.55, 13.08), (77.55, 13.01)]
     )
 
-    aoi2_polygon = Polygon(
-        [(77.56, 13.02), (77.59, 13.02), (77.59, 13.09), (77.56, 13.09), (77.56, 13.02)]
-    )
-
-    # get total bounds of all polygons above for stac search and stac index creation
-    bbox = aoi1_polygon.union(aoi2_polygon).bounds
+    bbox = aoi1_polygon.bounds
 
-    # 2. List existing collections
     print("1. Available Collections")
     print("----------------------")
     collections = Rasteret.list_collections(workspace_dir=workspace_dir)
@@ -39,9 +33,8 @@ def main():
             f"- {c['name']}: {c['data_source']}, {c['date_range']}, {c['size']} scenes"
         )
 
-    # 3. Try loading existing collection or create new
     try:
-        processor = Rasteret.load_collection(f"{custom_name}_202401-03_landsat")
+        processor = Rasteret.load_collection(f"{custom_name}_202403-03_landsat")
     except ValueError:
         print("\n2. Creating New Collection")
         print("-------------------------")
@@ -63,9 +56,7 @@ def main():
 
     # Calculate NDVI using xarray operations
     ds = processor.get_xarray(
-        # pass multiple geometries not its union bounds
-        # for separate processing of each geometry
-        geometries=[aoi1_polygon, aoi2_polygon],
+        geometries=[aoi1_polygon],
         bands=["B4", "B5"],
         cloud_cover_lt=20,
     )
@@ -77,18 +68,16 @@ def main():
     ndvi = (ds.B5 - ds.B4) / (ds.B5 + ds.B4)
     ndvi_ds = xr.Dataset(
         {"NDVI": ndvi},
-        coords=ds.coords,  # Preserve coordinates including CRS
-        attrs=ds.attrs,  # Preserve metadata
+        coords=ds.coords,
+        attrs=ds.attrs,
     )
 
     print("\nNDVI dataset:")
     print(ndvi_ds)
 
-    # Create output directory
     output_dir = Path("ndvi_results")
     output_dir.mkdir(exist_ok=True)
 
-    # Save per geometry, give prefix for output files in this case "ndvi"
     output_files = save_per_geometry(
         ndvi_ds, output_dir, file_prefix="ndvi", data_var="NDVI"
     )

setup.py

@@ -8,7 +8,7 @@
 
 setup(
     name="rasteret",
-    version="0.1.9",
+    version="0.1.10",
     author="Sidharth Subramaniam",
     author_email="sid@terrafloww.com",
     description="Fast and efficient access to Cloud-Optimized GeoTIFFs (COGs)",

src/rasteret/__init__.py

@@ -20,4 +20,10 @@ def version():
 __version__ = version()
 
 
-__all__ = ["Collection", "Rasteret", "CloudConfig", "AWSProvider", "DataSources"]
+__all__ = [
+    "Collection",
+    "Rasteret",
+    "CloudConfig",
+    "AWSProvider",
+    "DataSources",
+]

src/rasteret/core/collection.py

@@ -7,7 +7,6 @@
 import pyarrow.parquet as pq
 import pyarrow.dataset as ds
 import pandas as pd
-import geopandas as gpd
 
 from shapely.geometry import Polygon
 from pathlib import Path