@@ -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