From 63045d0f4f09e522ec4bd94768ec5373284a4e8c Mon Sep 17 00:00:00 2001
From: Ty Dickinson <tdickinson@salientpredictions.com>
Date: Fri, 12 Sep 2025 12:25:39 -0500
Subject: [PATCH] use time_dim arg, dont assume time

---
 src/xarray_regrid/regrid.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/xarray_regrid/regrid.py b/src/xarray_regrid/regrid.py
index 6b9942d..b2ed389 100644
--- a/src/xarray_regrid/regrid.py
+++ b/src/xarray_regrid/regrid.py
@@ -292,7 +292,7 @@ def validate_input(
     time_dim: str | None,
 ) -> xr.Dataset:
     if time_dim is not None and time_dim in ds_target_grid.coords:
-        ds_target_grid = ds_target_grid.isel(time=0).reset_coords()
+        ds_target_grid = ds_target_grid.isel({time_dim: 0}).reset_coords()
 
     if len(set(data.dims).intersection(set(ds_target_grid.dims))) == 0:
         msg = (