U %|e"@sddlmZddlmZddlmZddlmZddlm Z dd d d Z dd d Z ddZ ddZ ddZGdddZddddZdS)wraps)issparse)check_pandas_support) get_config) available_ifN)indexcCsrt|rtdt|rpandasdefaultz0output config must be 'default' or 'pandas' got dense)getattrrr )method estimatorZest_sklearn_output_configZ dense_configrrr_get_output_config?s  rcCs:t||}|ddkst|s"|St|t|dd|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. rrr N)rr r )r_auto_wrap_is_configuredrrget_feature_names_out)rrZoriginal_inputrZ output_configrrr_wrap_data_with_containeras  r cstfdd}|S)z@Wrapper used by `_SetOutputMixin` to automatically wrap methods.csJ||f||}t|tr.wrappedr)r'rr(rr&r_wrap_method_outputs r)cCs 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)rsethasattr)rauto_wrap_output_keysrrrrs rcs8eZdZdZdfdd 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. r+c 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. r+r1Nr=zUnable to configure output for z' because `set_output` is not available.r0)r-r r=)rr+Zset_output_for_transformrrr_safe_set_outputs    rC)N) functoolsr scipy.sparserr_configrZ _available_ifr rrr r)rr/rCrrrrs      5 "& I