core-lens 0.1.dev121__tar.gz → 0.1.dev124__tar.gz

{core_lens-0.1.dev121 → core_lens-0.1.dev124}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: core-lens
-Version: 0.1.dev121
+Version: 0.1.dev124
 Summary: Query, analyse, and visualise CoreStack's microwatershed and Earth science data through a clean, composable Python API.
 Project-URL: Homepage, https://github.com/ApoorvaKashyap/core-lens
 Project-URL: Issues, https://github.com/ApoorvaKashyap/core-lens/issues

{core_lens-0.1.dev121 → core_lens-0.1.dev124}/examples/demo_mws.py

@@ -30,11 +30,18 @@ from __future__ import annotations
 
 import polars as pl
 import shapely.geometry as sgeom
-from bokeh.io import show, save
+from bokeh.io import save
+from bokeh.resources import CDN
 
 from core_lens import AoI, SeasonConfig
 from core_lens.entities import MWSEntity
 from core_lens.export import geoparquet
+from core_lens.base.namespaces.stats import (
+    CorrelateMethod,
+    SimilarityMethod,
+    TestMethod,
+)
+from core_lens.base.view import Season
 
 DATA_ROOT = "/home/akashyap/Finals"
 
@@ -128,7 +135,7 @@ print("Resolution            :", result_annual.resolution)
 print("Has geometry          :", result_annual.has_geometry)  # False by default
 
 # Fortnightly data for a specific season and year
-result_fn = view_barmer.between(season="kharif", year=2021).fortnightly
+result_fn = view_barmer.between(season=Season.KHARIF, year=2021).fortnightly
 print("\nFortnightly columns (first 10):", result_fn.columns[:10])
 
 
@@ -171,7 +178,7 @@ print(desc_annual_by_entity.df().head(5))
 # Pairwise correlations on fortnightly data
 corr = result_fn.stats.correlate(
     columns=["df_precipitation", "df_et"],
-    method="pearson",
+    method=CorrelateMethod.PEARSON,
     across="entity",
 )
 print("\nCorrelation (Precipitation vs ET):")
@@ -184,7 +191,7 @@ print(corr.df())
 test_result = result_static_derived.stats.test(
     column="di_avg_dryspe",
     groups="district",
-    method="mann-whitney",
+    method=TestMethod.MANN_WHITNEY,
 )
 print("\nHypothesis test — dryspell by district:")
 print(test_result.df())
@@ -213,7 +220,7 @@ sim = result_static_derived.stats.similarity(
         "area_in_ha": None,
         "di_avg_dryspe": None,
     },
-    method="euclidean",
+    method=SimilarityMethod.EUCLIDEAN,
     top_n=5,
 )
 print(f"\nMost similar MWS to {target_id}:")
@@ -254,7 +261,7 @@ except Exception:
 
 try:
     line_fig = result_annual.plot.timeseries(x="year", y="dw_et")
-    save(line_fig, filename="mws_et_trend.html")
+    save(line_fig, filename="mws_et_trend.html", resources=CDN, title="MWS ET Trend")
     print("Saved line plot to mws_et_trend.html")
 except Exception:
     pass
@@ -263,7 +270,12 @@ except Exception:
 # ── 21. Plot — scatter ───────────────────────────────────────────────────────
 try:
     scatter_fig = result_static_derived.plot.scatter(x="area_km2", y="dryspell_score")
-    save(scatter_fig, filename="mws_scatter.html")
+    save(
+        scatter_fig,
+        filename="mws_scatter.html",
+        resources=CDN,
+        title="MWS Scatter Distribution",
+    )
 except Exception:
     pass
 
@@ -290,7 +302,7 @@ print("GeoParquet written: output_mws.geoparquet")
 
 final = (
     aoi.mws.where(ba_name="Barmer")
-    .between(season="kharif", year=2021)
+    .between(season=Season.KHARIF, year=2021)
     .fortnightly.derive(
         "et_ratio", (pl.col("df_et") / pl.col("df_precipitation")).round(3)
     )

{core_lens-0.1.dev121 → core_lens-0.1.dev124}/examples/demo_tehsil.py

@@ -33,7 +33,7 @@ from core_lens.base.namespaces.stats import (
     TestMethod,
 )
 import shapely.geometry as sgeom
-from bokeh.io import show, save
+from bokeh.io import show
 
 
 DATA_ROOT = "data/"

{core_lens-0.1.dev121 → core_lens-0.1.dev124}/src/core_lens/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.1.dev121'
-__version_tuple__ = version_tuple = (0, 1, 'dev121')
+__version__ = version = '0.1.dev124'
+__version_tuple__ = version_tuple = (0, 1, 'dev124')
 
 __commit_id__ = commit_id = None

{core_lens-0.1.dev121 → core_lens-0.1.dev124}/src/core_lens/base/namespaces/plot.py

@@ -144,12 +144,33 @@ def _wkb_to_arrow_table(
     idx = arrow_table.schema.get_field_index(geom_col)
     schema = arrow_table.schema.remove(idx)
 
+    import json
+
+    arr_geo = ac.Array(geo_col)
+    ac_geo = ac.ChunkedArray([arr_geo])
+    geom_field = arr_geo.field.with_name(geom_col)
+
+    ext_meta = dict(geom_field.metadata or {})
+    geo_meta = {}
+    if b"ARROW:extension:metadata" in ext_meta:
+        try:
+            geo_meta = json.loads(ext_meta[b"ARROW:extension:metadata"].decode("utf-8"))
+        except Exception:
+            pass
+    geo_meta["crs"] = "EPSG:4326"
+    ext_meta[b"ARROW:extension:metadata"] = json.dumps(geo_meta).encode("utf-8")
+    new_geom_field = geom_field.with_metadata(ext_meta)
+
+    new_fields = [
+        ac.Field.from_arrow(schema.field(i)) for i in range(len(schema.names))
+    ]
+    new_fields.insert(idx, new_geom_field)
+    new_schema = ac.Schema(new_fields)
+
     tbl_cols = [ac.ChunkedArray(arrow_table.column(c)) for c in schema.names]
-    tbl_cols.insert(idx, ac.ChunkedArray([ac.Array(geo_col)]))
-    names = list(schema.names)
-    names.insert(idx, geom_col)
+    tbl_cols.insert(idx, ac_geo)
 
-    tbl = ac.Table.from_arrays(tbl_cols, names=names)
+    tbl = ac.Table.from_arrays(tbl_cols, schema=new_schema)
     return tbl
 
 
@@ -447,8 +468,9 @@ class PlotNamespace:
                     size=5,
                 )
 
-            fig.legend.location = "top_left"
-            fig.legend.click_policy = "hide"
+            if fig.legend:
+                fig.legend.location = "top_left"
+                fig.legend.click_policy = "hide"
             _apply_theme(fig, self.result, title)
             return fig
 
@@ -552,8 +574,9 @@ class PlotNamespace:
                     size=8,
                     alpha=0.7,
                 )
-            fig.legend.location = "top_left"
-            fig.legend.click_policy = "hide"
+            if fig.legend:
+                fig.legend.location = "top_left"
+                fig.legend.click_policy = "hide"
             _apply_theme(fig, self.result, "Scatter Distribution")
             return fig
 
@@ -621,8 +644,9 @@ class PlotNamespace:
                     alpha=0.6,
                     legend_label=str(entity),
                 )
-            fig.legend.location = "top_right"
-            fig.legend.click_policy = "hide"
+            if fig.legend:
+                fig.legend.location = "top_right"
+                fig.legend.click_policy = "hide"
             _apply_theme(fig, self.result, "Distribution")
             return fig