:j#HdZddlZddlZddlmZddlmZddlmZGdde Z Gdd e Z d Z Gd d ejZdd Zy)ztThis module contains the _EstimatorPrettyPrinter class used in BaseEstimator.__repr__ for pretty-printing estimatorsN) get_config) BaseEstimator) is_scalar_nanc"eZdZdZfdZxZS) KeyValTuplez@Dummy class for correctly rendering key-value tuples from dicts.c t|S)N)super__repr__)self __class__s j/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/sklearn/utils/_pprint.pyr zKeyValTuple.__repr__Psw!!)__name__ __module__ __qualname____doc__r __classcell__r s@r rrMsJ""rrceZdZdZy)KeyValTupleParamzEDummy class for correctly rendering key-value tuples from parameters.N)rrrrrr rrUsOrrcT|jd}tj|jjj Dcic]\}}||j c}}fd}|j Dcic]\}}|||s||c}}Scc}}wcc}}w)zgReturn dict (param_name: value) of parameters that were given to estimator with non-default values.Fdeepc|vry|tjk(ryt|tr|j|jk7ryt |t |k7rt |r t |syy)NTF)inspect_empty isinstancerr reprr)kv init_paramss r has_changedz$_changed_params..has_changedcsq K  q>W^^ + a 'AKK;q>;S;S,S 7d;q>* * +a. )mA.>r) get_paramsr signature__init__ parametersitemsdefault) estimatorparamsnameparamr#r r!r"s @r _changed_paramsr.[s ! !u ! -F##I$6$67BBK:E:K:K:MN;44&NK $\\^ ATQ{1a/@AqD AA#O" BsBB$B$ceZdZdZ dddddfd ZdZdZd Zd Zd Z d Z d Z e jjjZeeej"<e eej"<xZS)_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. NFT)compactindent_at_namen_max_elements_to_showct||||||||_|jrd|_t d|_||_y)N)r1print_changed_only)r r&_indent_at_name_indent_per_levelr _changed_onlyr3) r indentwidthdepthstreamr1r2r3r s r r&z _EstimatorPrettyPrinter.__init__sP vwG-   %&D "'\*>?'=#rc6t|||||jS)N changed_only) _safe_reprr9)r objectcontext maxlevelslevels r formatz_EstimatorPrettyPrinter.formats  GYD "FFIM7E   Src 2|j||||||dS)NTis_dict_format_params_or_dict_itemsr r(r=r:rNrCrEs r _format_dict_itemsz*_EstimatorPrettyPrinter._format_dict_itemss)00 669gud1  rc 2|j||||||dS)NFrQrSrUs r rLz&_EstimatorPrettyPrinter._format_paramss)00 669gue1  rc|j}||jz }dd|zz} d} |j|z dzx} } t|} t | }d}d}|s||j k(r |dy|dz }|} t | }|jr|\}}|j|||}|j|||}|s|jd }|rd nd }||z|z}t|d z}| |kr| } | r| } | |k\r| |z} || d} |||| | } |rtnt}|j|||||r|nd|||syy#t $rYywxYw#t $rd }| |z} | |z} YwxYw)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 . 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. ,  r5NFr, ...T': =, )rJr8_widthiternext StopIterationr3_compact_reprstriprKrr_format)r rBr=r:rNrCrErRrJdelimnldelimr; max_widthitnext_entlastn_itemsentr r!kreprvreprmiddlerepwclass_s r rTz4_EstimatorPrettyPrinter._format_params_or_dict_itemss $(((#,& KK&0144 &\ BxH$555g qLGC #8 }}1 1gu5 1gu5!KK,E!(cfnu,HqL19%E 'A:QJE%L E#J %LE$+[1AF LLs VV$YAwPU E    ! #Y& " #s$ E 7 E EEE43E4c|j}||jz }|jdkDr||jdz dzdd|zz}d} |j|z dzx} } t|} t | } d}d}|s||j k(r |dy|dz }| } t | } |jrI|j|||}t|d z}| |kr| } | r|} | |k\r| |z} || d } |||| |} |j||||r|nd|||syy#t $rYywxYw#t $rd }| |z} | |z} YwxYw) 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. r5rZrYr[NFrr\Tr`ra) rJr8rbrcrdrer3rfrgrKri)r r(r=r:rNrCrErJrjrkr;rlrmrnrorprqrurvs r _format_itemsz%_EstimatorPrettyPrinter._format_itemss  $(((  ! !A % 4))A-4 5#,& KK&0144 %[ BxH$555g qLGC #8 }}jjgu5HqL19%E 'A:QJE%L E#J %LE LLff4iQQV W7    ! #Y& " #s$- D' D6' D32D36E Ec |\}}|j|||} t|tr| jd} d} nd} |j | |j | |j |||t | zt | z|||y)z=Pretty printing for key-value tuples from dict or parameters.r]r_r^N)rgrrrhrJrirK) r rBr=r:rNrCrEr r!rurts r _pprint_key_val_tuplez-_EstimatorPrettyPrinter._pprint_key_val_tupleJs1jjGU+ f. /))C.CFF S V vvC(3v;6 7E r)r5PNN)rrrrr&rFrOrVrLrTryr{pprint PrettyPrinter _dispatchcopyrr rrrs@r r0r0us0h =#=*   =~.X`  &$$..335I(9Im$$%&;Ik""#rr0c\t|}|tjvrt|ddfSt |dd}t |t r|t jur|syt|}|r ||k\rdd||vfS||vrtj|ddfSd||<d}d} g} | j} |dz }t} t|jtj} | D]E\}}| ||||| \}}}| ||||| \}}}| |d ||xr|xr|}|s|sDd} G||=d d j| z|| fSt |t r|t jus"t |t"r|t"jurt |t r|sy d}nt%|dk(rd}n|syd}t|}|r||k\r |dzd||vfS||vrtj|ddfSd||<d}d} g} | j} |dz }|D]'}t||||| \}}}| ||sd}|s&d} )||=|d j| z|| fSt |t&r-t|}|r||k\r|j(dd||vfS||vrtj|ddfSd||<d}d} |r t+|}n|j-d}g} | j} |dz }t} t|jtj} | D]T\}}| ||||| \}}}| ||||| \}}}| |j/dd||xr|xr|}|s|sSd} V||=|j(dd j| d|| fSt|}||xr|j1d dfS)zMSame as the builtin _safe_repr, with added support for Estimator objects.TFr N)z{}TFz{...}r5)keyr?r^z{%s}ra)z[]TFz[%s]z(%s,))z()TFz(%s)z...z(...)rr]r_rHrI<)typer}_builtin_scalarsrgetattr issubclassdictr id _recursionappendrArMr( _safe_tuplejoinlisttuplerKrrr.r$rh startswith)rBrCrDrEr@typrobjidreadable recursive componentsrsafereprr(r r!rr kreadablekrecurrs vreadablevrecurrFoorepr oreadableorecurr+rus r rArAbsD v,C f%%%F|T5((Z&A#tdmm!3$6  )+E5G#33 3 G $$V,eT9 9  ""  v||~6+=+=> !DAq'/7Iu<( $E9f(07Iu<( $E9f ue, -;I;)H  ! EN *--xBB3!t}}"431#6 c4 (F [A F(F6  )+E>5%7*:: : G $$V,eT9 9  ""   !A'17Iu<( $E9f 5M   ! EN *--xBB#}%6  )+ll^5)5%72BB B G $$V,eT9 9 $V,F&&E&2F ""  v||~6+=+=> !DAq'/7Iu<( $E9f(07Iu<( $E9f ekk#.6 7;I;)H  ! ENLL$))J*?@(IVV v,C 0S^^C005 88r)F)rrr}sklearn._configr sklearn.basersklearn.utils._missingrrrrr.r~r0rArrr rsU9H &&0"%" { B4j