Wrap arr lists in np.typed.List in nanops_numba

eoincondron · eoincondron · commit 36afc0f3c784 · 2025-09-26T13:05:52.000+01:00
diff --git a/pandas/core/nanops_numba.py b/pandas/core/nanops_numba.py
@@ -513,6 +513,7 @@ def _chunk_arr_into_arr_list(
             chunk for row in mask_list for chunk in np.array_split(row, n_threads)
         ]
 
+    arr_list, mask_list = NumbaList(arr_list), NumbaList(mask_list)
     return arr_list, mask_list, final_length
 
 

Original file line number	Diff line number	Diff line change
`@@ -513,6 +513,7 @@ def _chunk_arr_into_arr_list(`
`513`	`513`	`chunk for row in mask_list for chunk in np.array_split(row, n_threads)`
`514`	`514`	`]`
`515`	`515`
	`516`	`+ arr_list, mask_list = NumbaList(arr_list), NumbaList(mask_list)`
`516`	`517`	`return arr_list, mask_list, final_length`
`517`	`518`
`518`	`519`