from pint_xarray import unit_registry as ureg

@ureg.wraps(ret=None, args=["Newtons / second^2"])
def jpl_trajectory_code(acceleration):
    # do some rocket science
    ...


lockheed_acceleration_value = xr.DataArray(5).pint.quantify("pounds / second^2")

jpl_trajectory_code(lockheed_acceleration_value)

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-25-c85d33b3d64a> in <module>
----> 1 jpl_trajectory_code(lockheed_acceleration_value)

~/miniconda3/envs/py38-mamba/lib/python3.8/site-packages/pint/registry_helpers.py in wrapper(*values, **kw)
    263             # In principle, the values are used as is
    264             # When then extract the magnitudes when needed.
--> 265             new_values, values_by_name = converter(ureg, values, strict)
    266 
    267             result = func(*new_values, **kw)

~/miniconda3/envs/py38-mamba/lib/python3.8/site-packages/pint/registry_helpers.py in _converter(ureg, values, strict)
    147                         )
    148                     else:
--> 149                         raise ValueError(
    150                             "A wrapped function using strict=True requires "
    151                             "quantity or a string for all arguments with not None units. "

ValueError: A wrapped function using strict=True requires quantity or a string for all arguments with not None units. (error found for Newtons / second ** 2, <xarray.DataArray ()>
<Quantity(5, 'pound / second ** 2')>)

lockheed_acceleration_value = pint.Quantity(5, units="pounds / second^2")

In [32]: lockheed_acceleration_value
Out[32]: array(5) <Unit('pound / second ** 2')>

In [33]: jpl_trajectory_code(lockheed_acceleration_value)
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-33-c85d33b3d64a> in <module>
----> 1 jpl_trajectory_code(lockheed_acceleration_value)

~/miniconda3/envs/py38-mamba/lib/python3.8/site-packages/pint/registry_helpers.py in wrapper(*values, **kw)
    263             # In principle, the values are used as is
    264             # When then extract the magnitudes when needed.
--> 265             new_values, values_by_name = converter(ureg, values, strict)
    266 
    267             result = func(*new_values, **kw)

~/miniconda3/envs/py38-mamba/lib/python3.8/site-packages/pint/registry_helpers.py in _converter(ureg, values, strict)
    147                         )
    148                     else:
--> 149                         raise ValueError(
    150                             "A wrapped function using strict=True requires "
    151                             "quantity or a string for all arguments with not None units. "

ValueError: A wrapped function using strict=True requires quantity or a string for all arguments with not None units. (error found for Newtons / second ** 2, 5 pound / second ** 2)