fix: map geoarrow to geoarrow-types

[kirangadhave]

📝 Summary

Map the geoarrow import to the geoarrow-types PyPI package so the missing-package alert suggests the installable name.

Run the geometry fixture corpus in CI. Smoke the corpus through ui.table search and column summaries for GeoPandas, GeoArrow, and DuckDB. Add a subprocess check that a hand-built GeoArrow WKB table still types as geometry when geopandas and shapely cannot be imported.

Keep GeoArrow field metadata when default selection="multi" adds a row-id column. PyArrow now uses native Table.add_column instead of narwhals with_row_index, which dropped extension metadata.

Last PR of the geometry stack for MO-6362

Closes MO-7327

Made with Cursor

[vercel]

The latest updates on your projects. Learn more about Vercel for GitHub.

Project	Deployment	Actions	Updated (UTC)
marimo-docs	Ready	Preview	Aug 26, 2026 6:12am

[github-actions]

All contributors have signed the CLA ✍️ ✅
_{Posted by the CLA Assistant Lite bot.}

[cubic-dev-ai]

All reported issues were addressed across 4 files

Architecture diagram

sequenceDiagram
    participant User as User/Frontend
    participant Kernel as Marimo Runtime
    participant PkgMgr as module_name_to_pypi_name
    participant TableUI as ui.table
    participant Select as selection.py
    participant TM as TableManager
    participant Data as PyArrow/GeoData

    Note over User,Data: Flow 1: Missing Package Discovery
    User->>Kernel: import geoarrow
    Kernel->>Kernel: Import fails
    Kernel->>PkgMgr: lookup("geoarrow")
    PkgMgr-->>Kernel: NEW: "geoarrow-types"
    Kernel-->>User: Suggest "pip install geoarrow-types"

    Note over User,Data: Flow 2: Table Initialization with Row Selection
    User->>TableUI: Create table (selection=True)
    TableUI->>Select: add_selection_column(data)
    
    alt Data is PyArrow Table
        Select->>Data: CHANGED: native.add_column(index_col)
        Note right of Data: Preserves ARROW:extension metadata
    else Other DataFrames (Pandas/Polars)
        Select->>Data: Use Narwhals/Native insert
    end
    Select-->>TableUI: Data with row-id + preserved metadata
    
    TableUI->>TM: get_field_types()
    
    Note over TM,Data: Flow 3: Geometry Type Inference
    TM->>Data: Inspect schema/metadata
    
    alt Has GeoArrow Extension Metadata
        Data-->>TM: b"geoarrow.wkb" / b"geoarrow.wkt"
        TM-->>TableUI: ("geometry", "geoarrow.wkb")
    else Has GeoPandas/Shapely
        Data-->>TM: Geometry column
        TM-->>TableUI: ("geometry", "geometry")
    else DuckDB Spatial
        Data-->>TM: GEOMETRY type
        TM-->>TableUI: ("geometry", "GEOMETRY")
    end

    TableUI-->>User: Render table with Geometry formatting

Loading

<sub>Reply with feedback, questions, or to request a fix.

Re-trigger cubic</sub>

[Copilot]

Pull request overview

This PR improves the geometry-table “stack” by (1) mapping import geoarrow to the installable PyPI package name geoarrow-types for missing-module guidance, and (2) hardening geometry handling/tests—especially around GeoArrow extension metadata preservation when adding the default multi-selection row-id column.

Changes:

• Map the geoarrow module name to the geoarrow-types PyPI distribution for missing-package suggestions.
• Preserve GeoArrow extension field metadata when adding the selection row-id column for PyArrow-backed tables.
• Expand/enable geometry corpus coverage in CI with host-level smoke tests and an isolated subprocess test that blocks geopandas/shapely imports.

Reviewed changes

Copilot reviewed 4 out of 4 changed files in this pull request and generated no comments.

File	Description
tests/_plugins/ui/_impl/tables/test_geometry.py	Adds selection-metadata regression coverage, corpus smoke tests (search + summaries), and a subprocess test ensuring GeoArrow typing/formatting works without geopandas/shapely.
tests/_plugins/ui/_impl/tables/test_geometry_fixtures.py	Removes the CI skip gate so geometry characterization tests can run in CI (still guarded by pytest.mark.requires).
marimo/_runtime/packages/module_name_to_pypi_name.py	Adds geoarrow -> geoarrow-types mapping for install-name suggestions.
marimo/_plugins/ui/_impl/tables/selection.py	Uses native pyarrow.Table.add_column for PyArrow data to avoid losing GeoArrow extension metadata when adding the selection row-id column.

---

💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.