o
    tBhTH                     @   s   d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl	m
Z
 G d	d
 d
eZG dd deZdd ZG dd dejZdddZdS )ztThis module contains the _EstimatorPrettyPrinter class used in
BaseEstimator.__repr__ for pretty-printing estimators    N)OrderedDict   )BaseEstimator)
get_config   )is_scalar_nanc                       s    e Zd ZdZ fddZ  ZS )KeyValTuplez@Dummy class for correctly rendering key-value tuples from dicts.c                    s
   t   S N)super__repr__)self	__class__ l/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/utils/_pprint.pyr   N   s   
zKeyValTuple.__repr__)__name__
__module____qualname____doc__r   __classcell__r   r   r   r   r   K   s    r   c                   @   s   e Zd ZdZdS )KeyValTupleParamzEDummy class for correctly rendering key-value tuples from parameters.N)r   r   r   r   r   r   r   r   r   S   s    r   c                    s\   | j dd}t| jd| j}t|jdd  D fdd  fdd| D S )	zgReturn dict (param_name: value) of parameters that were given to
    estimator with non-default values.Fdeepdeprecated_originalc                 S   s   i | ]\}}||j qS r   )default).0nameparamr   r   r   
<dictcomp>`   s    z#_changed_params.<locals>.<dictcomp>c                    sl   |  vrdS  |  t jkrdS t|tr|j |  jkrdS t|t |  kr4t |  r2t|s4dS dS )NTF)inspect_empty
isinstancer   r   reprr   )kv)init_paramsr   r   has_changedb   s   
z$_changed_params.<locals>.has_changedc                    s    i | ]\}} ||r||qS r   r   )r   r#   r$   )r&   r   r   r   q   s     )
get_paramsgetattr__init__r   	signature
parametersitems)	estimatorparams	init_funcr   )r&   r%   r   _changed_paramsY   s   r0   c                       s   e Zd ZdZ				ddddd fdd	Zd
d Zdd Zdd Zdd Zdd Z	dd Z
dd Zejj Zeeej< eeej<   ZS )_EstimatorPrettyPrintera
  Pretty Printer class for estimator objects.

    This extends the pprint.PrettyPrinter class, because:
    - we need estimators to be printed with their parameters, e.g.
      Estimator(param1=value1, ...) which is not supported by default.
    - the 'compact' parameter of PrettyPrinter is ignored for dicts, which
      may lead to very long representations that we want to avoid.

    Quick overview of pprint.PrettyPrinter (see also
    https://stackoverflow.com/questions/49565047/pprint-with-hex-numbers):

    - the entry point is the _format() method which calls format() (overridden
      here)
    - format() directly calls _safe_repr() for a first try at rendering the
      object
    - _safe_repr formats the whole object recursively, only calling itself,
      not caring about line length or anything
    - back to _format(), if the output string is too long, _format() then calls
      the appropriate _pprint_TYPE() method (e.g. _pprint_list()) depending on
      the type of the object. This where the line length and the compact
      parameters are taken into account.
    - those _pprint_TYPE() methods will internally use the format() method for
      rendering the nested objects of an object (e.g. the elements of a list)

    In the end, everything has to be implemented twice: in _safe_repr and in
    the custom _pprint_TYPE methods. Unfortunately PrettyPrinter is really not
    straightforward to extend (especially when we want a compact output), so
    the code is a bit convoluted.

    This class overrides:
    - format() to support the changed_only parameter
    - _safe_repr to support printing of estimators (for when they fit on a
      single line)
    - _format_dict_items so that dict are correctly 'compacted'
    - _format_items so that ellipsis is used on long lists and tuples

    When estimators cannot be printed on a single line, the builtin _format()
    will call _pprint_estimator() because it was registered to do so (see
    _dispatch[BaseEstimator.__repr__] = _pprint_estimator).

    both _format_dict_items() and _pprint_estimator() use the
    _format_params_or_dict_items() method that will format parameters and
    key-value pairs respecting the compact parameter. This method needs another
    subroutine _pprint_key_val_tuple() used when a parameter or a key-value
    pair is too long to fit on a single line. This subroutine is called in
    _format() and is registered as well in the _dispatch dict (just like
    _pprint_estimator). We had to create the two classes KeyValTuple and
    KeyValTupleParam for this.
    r   P   NFT)compactindent_at_namen_max_elements_to_showc                   s>   t  j|||||d || _| jrd| _t d | _|| _d S )N)r3   r   print_changed_only)r
   r)   _indent_at_name_indent_per_levelr   _changed_onlyr5   )r   indentwidthdepthstreamr3   r4   r5   r   r   r   r)      s   
z _EstimatorPrettyPrinter.__init__c                 C   s   t ||||| jdS )Nchanged_only)
_safe_reprr9   )r   objectcontext	maxlevelslevelr   r   r   format   s   z_EstimatorPrettyPrinter.formatc                 C   s   | |jjd  | jr|t|jj7 }| jrt|}n|jdd}tdd t	|
 D }| |
 |||d || | d d S )N(Fr   c                 s   s    | ]	\}}||fV  qd S r	   r   )r   r   valr   r   r   	<genexpr>   s    z<_EstimatorPrettyPrinter._pprint_estimator.<locals>.<genexpr>r   ))writer   r   r7   lenr9   r0   r'   r   sortedr,   _format_params)r   rA   r=   r:   	allowancerB   rD   r.   r   r   r   _pprint_estimator   s   
z)_EstimatorPrettyPrinter._pprint_estimatorc              	   C      | j ||||||ddS )NTis_dict_format_params_or_dict_itemsr   r,   r=   r:   rN   rB   rD   r   r   r   _format_dict_items      z*_EstimatorPrettyPrinter._format_dict_itemsc              	   C   rP   )NFrQ   rS   rU   r   r   r   rM      rW   z&_EstimatorPrettyPrinter._format_paramsc                 C   s  |j }|| j7 }dd|  }	d}
| j| d  }}t|}zt|}W n
 ty-   Y dS w d}d}|s|| jkr?|d dS |d7 }|}zt|}W n ty^   d	}||8 }||8 }Y nw | jr|\}}| |||}| |||}|s{|	d
}|rdnd}|| | }t
|d }||k r|}|
r|	}
||kr||8 }||
 d}
|| q2||
 |	}
|rtnt}| |||||r|nd|| |r4dS dS )a-  Format dict items or parameters respecting the compact=True
        parameter. For some reason, the builtin rendering of dict items doesn't
        respect compact=True and will use one line per key-value if all cannot
        fit in a single line.
        Dict items will be rendered as <'key': value> while params will be
        rendered as <key=value>. The implementation is mostly copy/pasting from
        the builtin _format_items().
        This also adds ellipsis if the number of items is greater than
        self.n_max_elements_to_show.
        ,
  r   NFr   , ...T': =r   , )rJ   r8   _widthiternextStopIterationr5   _compact_reprstriprK   r   r   _format)r   rA   r=   r:   rN   rB   rD   rR   rJ   delimnldelimr;   	max_widthitnext_entlastn_itemsentr#   r$   kreprvreprmiddlerepwclass_r   r   r   rT      sh   


z4_EstimatorPrettyPrinter._format_params_or_dict_itemsc                 C   sh  |j }|| j7 }| jdkr|| jd d  dd|  }d}	| j| d  }
}t|}zt|}W n
 ty;   Y dS w d}d}|s|| jkrM|d dS |d7 }|}zt|}W n tyl   d	}||8 }|
|8 }
Y nw | jr| |||}t	|d
 }|
|k r|}
|	r|}	|
|kr|
|8 }
||	 d}	|| q@||	 |}	| 
||||r|nd|| |rBdS dS )zFormat the items of an iterable (list, tuple...). Same as the
        built-in _format_items, with support for ellipsis if the number of
        elements is greater than self.n_max_elements_to_show.
        r   rY   rX   rZ   NFr   r[   Tr   r_   )rJ   r8   r`   ra   rb   rc   r5   rd   re   rK   rg   )r   r,   r=   r:   rN   rB   rD   rJ   rh   ri   r;   rj   rk   rl   rm   rn   ro   rs   rt   r   r   r   _format_items  sZ   


z%_EstimatorPrettyPrinter._format_itemsc                 C   sp   |\}}|  |||}	t|tr|	d}	d}
nd}
||	 ||
 | |||t|	 t|
 ||| dS )z=Pretty printing for key-value tuples from dict or parameters.r\   r^   r]   N)re   r!   r   rf   rJ   rg   rK   )r   rA   r=   r:   rN   rB   rD   r#   r$   rs   rr   r   r   r   _pprint_key_val_tupleK  s   



z-_EstimatorPrettyPrinter._pprint_key_val_tuple)r   r2   NN)r   r   r   r   r)   rE   rO   rV   rM   rT   rv   rw   pprintPrettyPrinter	_dispatchcopyr   r   r   r   r   r   r   r   r1   t   s*    4?0
r1   Fc                 C   s  t | }|tjv rt| ddfS t|dd}t|tr|tju r| s$dS t| }|r5||kr5dd||v fS ||v rAt	| ddfS d||< d}d}	g }
|
j
}|d7 }t}t|  tjd}|D ]0\}}||||||d	\}}}||||||d	\}}}|d
||f  |o|o|}|s|rd}	q_||= dd|
 ||	fS t|tr|tju st|tr+|tju r+t|tr| sdS d}nt| dkrd}n| sdS d}t| }|r||kr|d d||v fS ||v rt	| ddfS d||< d}d}	g }
|
j
}|d7 }| D ]}t|||||d	\}}}|| |sd}|rd}	q ||= |d|
 ||	fS t|trt| }|rD||krDdd||v fS ||v rQt	| ddfS d||< d}d}	|rat| }n| jdd}g }
|
j
}|d7 }t}t| tjd}|D ]8\}}||||||d	\}}}||||||d	\}}}|d|d|f  |o|o|}|s|rd}	q}||= d|jd|
f ||	fS t| }||o|d dfS )zMSame as the builtin _safe_repr, with added support for Estimator
    objects.TFr   N)z{}TFz{...}r   )keyr>   z%s: %sz{%s}r_   )z[]TFz[%s]z(%s,))z()TFz(%s)z...r   z%s=%sr\   z%s(%s)<)typerx   _builtin_scalarsr"   r(   
issubclassdictr   id
_recursionappendr@   rL   r,   _safe_tuplejoinlisttuplerK   r   r0   r'   rf   r   
startswith)rA   rB   rC   rD   r?   typrobjidreadable	recursive
componentsr   safereprr,   r#   r$   rp   	kreadablekrecurrq   	vreadablevrecurrE   oorepr	oreadableorecurr.   rs   r   r   r   r@   c  s   








r@   )F)r   r   rx   collectionsr   baser   _configr   rZ   r   r   r   r   r0   ry   r1   r@   r   r   r   r   <module>   s    A p