Update type hints for curvefit p0 and bounds

mgunyho · Marton Gunyho · commit 6e4aef7c7d20 · 2023-05-16T10:34:15.000+03:00
diff --git a/xarray/core/dataarray.py b/xarray/core/dataarray.py
@@ -6116,8 +6116,8 @@ def curvefit(
         func: Callable[..., Any],
         reduce_dims: Dims = None,
         skipna: bool = True,
-        p0: dict[str, Any] | None = None,
-        bounds: dict[str, Any] | None = None,
+        p0: dict[str, float | DataArray] | None = None,
+        bounds: dict[str, tuple[float | DataArray, float | DataArray]] | None = None,
         param_names: Sequence[str] | None = None,
         kwargs: dict[str, Any] | None = None,
     ) -> Dataset:
diff --git a/xarray/core/dataset.py b/xarray/core/dataset.py
@@ -8560,8 +8560,8 @@ def curvefit(
         func: Callable[..., Any],
         reduce_dims: Dims = None,
         skipna: bool = True,
-        p0: dict[str, Any] | None = None,
-        bounds: dict[str, Any] | None = None,
+        p0: dict[str, float | DataArray] | None = None,
+        bounds: dict[str, tuple[float | DataArray, float | DataArray]] | None = None,
         param_names: Sequence[str] | None = None,
         kwargs: dict[str, Any] | None = None,
     ) -> T_Dataset:
