nan's picture
feat: return a list when the input is a list
f7df96a