Skip to content

Commit 347463e

Browse files
committed
Fix CI, update validate_values_for_numba params
1 parent 2703f86 commit 347463e

File tree

1 file changed

+18
-8
lines changed

1 file changed

+18
-8
lines changed

pandas/core/apply.py

Lines changed: 18 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,7 @@ def apply_raw_false(
271271
Series,
272272
)
273273

274-
engine_kwargs: dict[str, bool] | None = (
274+
engine_kwargs: dict[str, bool] = (
275275
decorator if isinstance(decorator, dict) else {}
276276
)
277277

@@ -301,17 +301,27 @@ def apply_raw_false(
301301
return DataFrame() if isinstance(data, DataFrame) else Series()
302302

303303
@staticmethod
304-
def validate_values_for_numba(df: DataFrame) -> None:
305-
for colname, dtype in df.dtypes.items():
306-
if not is_numeric_dtype(dtype):
304+
def validate_values_for_numba(obj: Series | DataFrame) -> None:
305+
if isinstance(obj, Series):
306+
if not is_numeric_dtype(obj.dtype):
307307
raise ValueError(
308-
f"Column {colname} must have numeric dtype. Found '{dtype}'."
308+
f"Series must have numeric dtype. Found '{dtype}'."
309309
)
310-
if is_extension_array_dtype(dtype):
310+
if is_extension_array_dtype(obj.dtype):
311311
raise ValueError(
312-
f"Column {colname} uses extension array dtype, "
313-
"not supported by Numba."
312+
f"Series uses extension array dtype, not supported by Numba."
314313
)
314+
else:
315+
for colname, dtype in obj.dtypes.items():
316+
if not is_numeric_dtype(dtype):
317+
raise ValueError(
318+
f"Column {colname} must have numeric dtype. Found '{dtype}'."
319+
)
320+
if is_extension_array_dtype(dtype):
321+
raise ValueError(
322+
f"Column {colname} uses extension array dtype, "
323+
"not supported by Numba."
324+
)
315325

316326
@staticmethod
317327
@functools.cache

0 commit comments

Comments
 (0)