move numbagg nanquantiles logic to nputils-module

xarray/core/nputils.py

@@ -206,6 +206,11 @@ def f(values, axis=None, **kwargs):
                 # to ddof=1 above.
                 if pycompat.mod_version("numbagg") < Version("0.7.0"):
                     kwargs.pop("ddof", None)
+                if name == "nanquantile":
+                    if kwargs.get("method", None) != "linear":
+                        return
+                    kwargs["quantiles"] = kwargs.pop("q")
+                    kwargs.pop("method", None)
                 return nba_func(values, axis=axis, **kwargs)
         if (
             _BOTTLENECK_AVAILABLE
@@ -285,3 +290,4 @@ def least_squares(lhs, rhs, rcond=None, skipna=False):
 nancumprod = _create_method("nancumprod")
 nanargmin = _create_method("nanargmin")
 nanargmax = _create_method("nanargmax")
+nanquantile = _create_method("nanquantile")

xarray/core/variable.py

@@ -1995,20 +1995,8 @@ def quantile(
 
             method = interpolation
 
-        using_numbagg = False
         if skipna or (skipna is None and self.dtype.kind in "cfO"):
-            if (
-                module_available("numbagg", minversion="0.5.1")
-                and OPTIONS["use_numbagg"] is True
-                and method == "linear"
-                and not isinstance(self.data, array_type("pint"))
-            ):
-                import numbagg
-
-                _quantile_func = numbagg.nanquantile
-                using_numbagg = True
-            else:
-                _quantile_func = np.nanquantile
+            _quantile_func = nputils.nanquantile
         else:
             _quantile_func = np.quantile
 
@@ -2030,10 +2018,7 @@ def _wrapper(npa, **kwargs):
 
         axis = np.arange(-1, -1 * len(dim) - 1, -1)
 
-        if using_numbagg:
-            kwargs = {"quantiles": q, "axis": axis}
-        else:
-            kwargs = {"q": q, "axis": axis, "method": method}
+        kwargs = {"q": q, "axis": axis, "method": method}
 
         result = apply_ufunc(
             _wrapper,