Jak używać numpy w opcjonalnym typowaniu

Question

Powiedzmy, że chcę utworzyć funkcję, która przyjmuje funkcję lambda (Callable) jako parametr, w którym funkcja lambda przyjmuje wektor jako dane wejściowe (zdefiniowane jako tablica numpy lub numpy matrix) i zwraca nową wartość wektor. Jak zadeklarować typ podpisu dla opcji Callable z typami numpy?

Moja początkowa próba wygląda mniej więcej tak:

def some_func(calc_new_vector: Callable[[np.array], np.array], ...other-params...) -> SomeType: 
    ...do stuff... 
    ...return... 

Jednak to powoduje wystąpienie błędu podczas uruchamiania interpretera:

TypeError: Callable[[arg, ...], result]: each arg must be a type. Got <built-in function array>. 

Answer 1

Confusingly, np.array jest funkcja przydatna do tworzenia numpy tablice. To nie jest rzeczywisty type utworzonych tablic.

Typ to np.ndarray.

Wymień np.array na np.ndarray. To powinno naprawić problem.