core-lens 0.1.dev160__tar.gz → 0.1.dev161__tar.gz

{core_lens-0.1.dev160 → core_lens-0.1.dev161}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: core-lens
-Version: 0.1.dev160
+Version: 0.1.dev161
 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.dev160 → core_lens-0.1.dev161}/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.dev160'
-__version_tuple__ = version_tuple = (0, 1, 'dev160')
+__version__ = version = '0.1.dev161'
+__version_tuple__ = version_tuple = (0, 1, 'dev161')
 
 __commit_id__ = commit_id = None

{core_lens-0.1.dev160 → core_lens-0.1.dev161}/src/core_lens/base/namespaces/plot.py

@@ -561,11 +561,19 @@ class PlotNamespace:
                             color=color,
                             size=5,
                         )
-            elif key_col in pdf.columns:
-                for i, entity in enumerate(pdf[key_col].unique()):
-                    row = pdf[pdf[key_col] == entity]
+            elif key_col in df.columns:
+                entities = df.get_column(key_col).unique().to_list()
+                for i, entity in enumerate(entities):
+                    entity_df = df.filter(pl.col(key_col) == entity)
+                    source_data = {}
+                    for c in hover_cols:
+                        if c in entity_df.columns:
+                            lst = entity_df.get_column(c).to_list()
+                            if entity_df.schema[c].is_numeric():
+                                lst = [float("nan") if v is None else v for v in lst]
+                            source_data[c] = lst
+                    source = ColumnDataSource(source_data)  # type: ignore[arg-type]
                     color = _color_for(i)
-                    source = _src(row)
                     fig.line(
                         x=x,
                         y=y_col,
@@ -676,8 +684,6 @@ class PlotNamespace:
             entities = unique_entities.limit(top_n).to_list()
             df = df.join(pl.DataFrame({key_col: entities}), on=key_col, how="inner")
 
-        df_pd = df.to_pandas()
-
         def _make_scatter(y_col: str) -> "BokehFigure":
             from bokeh.models import ColumnDataSource
 
@@ -689,11 +695,21 @@ class PlotNamespace:
                 tools="pan,wheel_zoom,box_zoom,reset,save,hover",
                 tooltips=[(c, f"@{{{c}}}") for c in hover_cols],
             )
-            for i, entity in enumerate(df_pd[key_col].unique()):
-                row = df_pd[df_pd[key_col] == entity]
-                source = ColumnDataSource(
-                    {c: row[c].tolist() for c in hover_cols if c in row.columns}
-                )
+
+            entities = df.get_column(key_col).unique().to_list()
+            for i, entity in enumerate(entities):
+                row = df.filter(pl.col(key_col) == entity)
+
+                # Replace None with NaN for numeric columns to prevent Bokeh JS rendering crashes
+                source_data = {}
+                for c in hover_cols:
+                    if c in row.columns:
+                        lst = row.get_column(c).to_list()
+                        if row.schema[c].is_numeric():
+                            lst = [float("nan") if v is None else v for v in lst]
+                        source_data[c] = lst
+
+                source = ColumnDataSource(source_data)  # type: ignore[arg-type]
                 fig.scatter(
                     x=x,
                     y=y_col,