remove unused import of ndarray