Come usare numpy nella digitazione opzionale

Question

Diciamo che voglio fare una funzione che prende una funzione lambda (Callable) come parametro dove la funzione lambda prende un vettore come input (definito come array numpy o numpy) e restituisce un nuovo vettore. Come si dichiara la firma del tipo per il chiamabile con tipi numpy?

Il mio primo tentativo di simile a questa:

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

Tuttavia, questo si traduce in un errore durante l'esecuzione l'interprete:

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

Answer 1

Confusamente, np.array è una funzione utile per creare array numpy. Non è l'attuale type degli array creati.

Il tipo è np.ndarray.

Quindi, sostituire np.array con np.ndarray. Questo dovrebbe risolvere il problema.