Невозможно проверить размерность с помощью Pint (элементы DataFrame) ⇐ Python

[Anonymous]

Я пытаюсь проверить размерность величин, передаваемых в функцию, с помощью пакета pint. В частности, у меня есть DataFrame Pandas, где типы данных столбцов задаются с помощью pint_pandas. Я хочу передать элементы этого DataFrame в свою функцию.
import pandas as pd
import pint
ureg = pint.UnitRegistry()
import pint_pandas

df = pd.DataFrame(
{
('range', 'km'): [200]
}
).pint.quantify(level=1)

@ureg.check(
'[length]',
)
def multiply_value(range):
return range * 2

multiply_value(df.iloc[0]['range'])

К сожалению, это возвращает ошибку:
{
"name": "DimensionalityError",
"message": "Cannot convert from '200 kilometer' ([length]) to 'a quantity of' ([length])",
"stack": "---------------------------------------------------------------------------
DimensionalityError Traceback (most recent call last)
Cell In[100], line 18
12 @ureg.check(
13 '[mass]',
14 )
15 def multiply_value(range):
16 return range * 2
---> 18 multiply_value(df.iloc[0]['range'])

File /opt/homebrew/Caskroom/miniconda/base/envs/aircraft_performance/lib/python3.13/site-packages/pint/registry_helpers.py:381, in check..decorator..wrapper(*args, **kwargs)
379 if not ureg.Quantity(value).check(dim):
380 val_dim = ureg.get_dimensionality(value)
--> 381 raise DimensionalityError(value, \"a quantity of\", val_dim, dim)
382 return func(*args, **kwargs)

DimensionalityError: Cannot convert from '200 kilometer' ([length]) to 'a quantity of' ([length])"
}

Если я вместо этого определяю количество пинты самостоятельно, это работает нормально:
>>> range = 200 * ureg.km
>>> multiply_value(range)
400 kilometer


Подробнее здесь: https://stackoverflow.com/questions/792 ... e-elements