Fix that removes data vars while preserving dimension coordinates so isel works correctly.

alxmrs · alxmrs · commit 1485a27a9671 · 2026-03-29T17:09:29.000-07:00
diff --git a/xarray_sql/reader.py b/xarray_sql/reader.py
@@ -271,7 +271,7 @@ def make_stream(
         else:
           # Only dimension coords requested — drop all data vars to avoid
           # loading them unnecessarily (e.g. for queries like SELECT lat, lon).
-          ds_block = ds[[]].isel(block)
+          ds_block = ds.drop_vars(list(ds.data_vars)).isel(block)
         batch_schema = pa.schema(
             [schema.field(name) for name in projection_names]
         )

Original file line number	Diff line number	Diff line change
`@@ -271,7 +271,7 @@ def make_stream(`
`271`	`271`	`else:`
`272`	`272`	`# Only dimension coords requested — drop all data vars to avoid`
`273`	`273`	`# loading them unnecessarily (e.g. for queries like SELECT lat, lon).`
`274`		`- ds_block = ds[[]].isel(block)`
	`274`	`+ ds_block = ds.drop_vars(list(ds.data_vars)).isel(block)`
`275`	`275`	`batch_schema = pa.schema(`
`276`	`276`	`[schema.field(name) for name in projection_names]`
`277`	`277`	`)`