U 9%e2&@sddlmZddlmZddlmZddlmZddlm Z dd d d Z dd d Z ddZ ddZ ddZGdddZddddZdS)wraps)issparse) get_config)check_pandas_support) available_ifN)indexcCstt|rtdt|rdefaultpandasz0output config must be 'default' or 'pandas' got dense)getattrrr )method estimatorZest_sklearn_output_configZ dense_configrrr_get_output_config?s  rcCsLt||}|ddkst|s"|Sdd}||r8|jnd}t|||jdS)aWrap output with container based on an estimator's or global config. Parameters ---------- method : {"transform"} Estimator's method to get container output for. data_to_wrap : {ndarray, dataframe} Data to wrap with container. original_input : {ndarray, dataframe} Original input of function. estimator : estimator instance Estimator with to get the output configuration from. Returns ------- output : {ndarray, dataframe} If the output config is "default" or the estimator is not configured for wrapping return `data_to_wrap` unchanged. If the output config is "pandas", return `data_to_wrap` as a pandas DataFrame. rrcSsRddl}t|drNt|drNz|jd}Wntk r@YdSXt||jSdS)zvReturn True if the X is a pandas dataframe. This is is backport from 1.4 in 1.3.1 for compatibility. rNr ZilocrF)syshasattrmodulesKeyErrorrr)Xr rrrr _is_pandas_dfs z0_wrap_data_with_container.._is_pandas_dfN)rr r )r_auto_wrap_is_configuredr rget_feature_names_out)rrZoriginal_inputrZ output_configr%r rrr_wrap_data_with_containeras r(cstfdd}|S)z@Wrapper used by `_SetOutputMixin` to automatically wrap methods.csj||f||}t|tr\t|d||f|dd}tt|drXt||S|St|||S)Nrr_make)rtupler(r!typer))selfr$argskwargsrZ return_tuplefrrrwrappeds  z$_wrap_method_output..wrappedr)r0rr1rr/r_wrap_method_outputsr2cCs t|dt}t|dod|kS)zReturn True if estimator is configured for auto-wrapping the transform method. `_SetOutputMixin` sets `_sklearn_auto_wrap_output_keys` to `set()` if auto wrapping is manually disabled. _sklearn_auto_wrap_output_keysr' transform)rsetr!)rauto_wrap_output_keysrrrr&s r&cs8eZdZdZdfdd ZeeddddZZS) _SetOutputMixina\Mixin that dynamically wraps methods to return container based on config. Currently `_SetOutputMixin` wraps `transform` and `fit_transform` and configures it based on `set_output` of the global configuration. `set_output` is only defined if `get_feature_names_out` is defined and `auto_wrap_output_keys` is the default value. r4c stjf|t|ts(|dks(td|dkrtd|d|j|dS)aSafely call estimator.set_output and error if it not available. This is used by meta-estimators to set the output for child estimators. Parameters ---------- estimator : estimator instance Estimator instance. transform : {"default", "pandas"}, default=None Configure output of the following estimator's methods: - `"transform"` - `"fit_transform"` If `None`, this operation is a no-op. Returns ------- estimator : estimator instance Estimator instance. r4r9NrDzUnable to configure output for z' because `set_output` is not available.r8)r!r rD)rr4Zset_output_for_transformrrr_safe_set_outputs    rJ)N) functoolsrZ scipy.sparser_configrrZ _available_ifr rrr(r2r&r7rJrrrrs      5 "7 I