Fix engine_kwargs handling in validate_values_for_numba_raw_false

arthurlw · arthurlw · commit e337cb8ff939 · 2025-06-23T21:59:49.000+07:00
diff --git a/pandas/core/apply.py b/pandas/core/apply.py
@@ -49,6 +49,7 @@
 import pandas.core.common as com
 from pandas.core.construction import ensure_wrapped_if_datetimelike
 from pandas.core.util.numba_ import (
+    get_jit_arguments,
     prepare_function_arguments,
 )
 
@@ -224,7 +225,7 @@ def apply(
             if data.empty:
                 return data.copy()  # mimic apply_empty_result()
             NumbaExecutionEngine.validate_values_for_numba_raw_false(
-                data, decorator if isinstance(decorator, dict) else {}
+                data, get_jit_arguments(decorator.engine_kwargs)
             )
 
             return NumbaExecutionEngine.apply_raw_false(
diff --git a/pandas/core/frame.py b/pandas/core/frame.py
@@ -10623,6 +10623,7 @@ def apply(
             numba = import_optional_dependency("numba")
             numba_jit = numba.jit(**engine_kwargs or {})
             numba_jit.__pandas_udf__ = NumbaExecutionEngine
+            numba_jit.engine_kwargs = engine_kwargs
             engine = numba_jit
 
         if engine is None or isinstance(engine, str):

Original file line number	Diff line number	Diff line change
`@@ -49,6 +49,7 @@`
`49`	`49`	`import pandas.core.common as com`
`50`	`50`	`from pandas.core.construction import ensure_wrapped_if_datetimelike`
`51`	`51`	`from pandas.core.util.numba_ import (`
	`52`	`+ get_jit_arguments,`
`52`	`53`	`prepare_function_arguments,`
`53`	`54`	`)`
`54`	`55`
`@@ -224,7 +225,7 @@ def apply(`
`224`	`225`	`if data.empty:`
`225`	`226`	`return data.copy() # mimic apply_empty_result()`
`226`	`227`	`NumbaExecutionEngine.validate_values_for_numba_raw_false(`
`227`		`- data, decorator if isinstance(decorator, dict) else {}`
	`228`	`+ data, get_jit_arguments(decorator.engine_kwargs)`
`228`	`229`	`)`
`229`	`230`
`230`	`231`	`return NumbaExecutionEngine.apply_raw_false(`