Как ввести функцию подсказки, чтобы она была совместима с numpy ⇐ Python

[Anonymous]

Исходный код example.py:
from typing import Union, Any
import numpy as np

Number = Union[int, float, np.floating[Any]]

def add_one(num: Number) -> Number:
return num + 1

inputs = [1, 2, 3]
outputs = [add_one(n) for n in inputs]

avg = np.mean(outputs)

Запустите mypy:
mypy example.py
src/example.py:14: error: Argument 1 to "mean" has incompatible type "List[Union[float, floating[Any]]]"; expected "Union[_SupportsArray[dtype[Any]], _NestedSequence[_SupportsArray[dtype[Any]]], bool, int, float, complex, str, bytes, _NestedSequence[Union[bool, int, float, complex, str, bytes]]]"
Found 1 error in 1 file (checked 1 source file)

Я могу изменить все на np.floating[Any], что устраняет проблему с numpy, но тогда мне придется привести примитивы к np.float32(...)
код>:
from typing import Any
import numpy as np

def add_one(num: np.floating[Any]) -> np.floating[Any]:
return num + 1

inputs = [1, 2, 3]
outputs = [add_one(np.float32(n)) for n in inputs]

avg = np.mean(outputs)

Есть ли правильный способ ввести подсказку функции add_one, чтобы ее выходные данные были совместимы с цифровыми функциями, такими как np.mean без нарушения совместимости с примитивными типами Python? Конечная цель — иметь возможность использовать его следующим образом:
inputs = [1, 2, 3]
outputs = [add_one(n) for n in inputs]
avg = np.mean(outputs)


Подробнее здесь: https://stackoverflow.com/questions/725 ... with-numpy