Allow numpy>=2. The application of the polynomial for the pressure wind relation always returns a numpy array with for 2.x, but returns the same type as the input for 1.x. I added the metpy wrapper to this part of the function so it will return xarray if xarray is passed and made sure the result is always a numpy array inside the function so it is consistent across 1.x and 2.x

Author: leosaffin

huracanpy/tc/_ace.py

@@ -2,6 +2,7 @@
 Module containing functions to compute ACE
 """
 
+import numpy as np
 from numpy.polynomial.polynomial import Polynomial
 import xarray as xr
 import pint
@@ -275,11 +276,16 @@ def get_pressure_wind_relation(pressure, wind=None, model=None, **kwargs):
             "Need to specify either wind or model to calculate pressure-wind relation"
         )
 
-    wind_from_fit = model(pressure)
+    wind_from_fit = _get_wind_from_model(pressure, model)
 
     return wind_from_fit, model
 
 
+@preprocess_and_wrap(wrap_like="pressure")
+def _get_wind_from_model(pressure, model):
+    return np.array(model(pressure))
+
+
 # Pre-determined pressure-wind relationships
 _z2021 = Polynomial([1.43290190e01, 5.68356519e-01, -1.05371378e-03])
 

pyproject.toml

@@ -9,7 +9,7 @@ authors = [
 description = "A python package for working with various forms of feature tracking data"
 requires-python = ">=3.8, <3.13"
 dependencies = [
-    "numpy<2",
+    "numpy",
     "xarray",
     "cftime",
     "parse",