@@ -271,7 +271,7 @@ def apply_raw_false(
271
271
Series ,
272
272
)
273
273
274
- engine_kwargs : dict [str , bool ] | None = (
274
+ engine_kwargs : dict [str , bool ] = (
275
275
decorator if isinstance (decorator , dict ) else {}
276
276
)
277
277
@@ -301,17 +301,27 @@ def apply_raw_false(
301
301
return DataFrame () if isinstance (data , DataFrame ) else Series ()
302
302
303
303
@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 ):
307
307
raise ValueError (
308
- f"Column { colname } must have numeric dtype. Found '{ dtype } '."
308
+ f"Series must have numeric dtype. Found '{ dtype } '."
309
309
)
310
- if is_extension_array_dtype (dtype ):
310
+ if is_extension_array_dtype (obj . dtype ):
311
311
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."
314
313
)
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
+ )
315
325
316
326
@staticmethod
317
327
@functools .cache
0 commit comments