nan's picture
fix: fix the bug when return_numpy is false
6bb8cf2