:jO0dZddlZddlZddlZddlZddlmZmZddlm Z m Z ddl m Z m Z mZmZddlmZddlZddlmZddlmZd ed efd Zd ed dfd Zded eeeeeeffdZ dejBd e"ee ffdZ#dejBd eee e"ee fffdZ$ d(dejBde%e"ee fdedeedfd e%e"ee ff dZ&ddddd e de deeeeefdeejBdee"ee fd df dZ'GddeZdd d!ed e%e fd"Z(dd d!ed ee fd#Z)dd d!ed efd$Z*dd d!ed ee fd%Z+dd d!ed&e d dfd'Z,y))z%Utilities used for parameter parsing.N)MutableMappingSequence)fields is_dataclass)AnyLiteralOptionalUnion)nn) AttributeDict)rank_zero_warnobjreturnc tj|y#tjttt f$rYywxYw)z"Tests if an object can be pickled.TF)pickledumps PickleErrorAttributeError RuntimeError TypeError)rs x/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/pytorch_lightning/utilities/parsing.py is_picklablers7 S    i Hs #>>hparamsc|jDcgc]\}}t|r|}}}|D]}td|d|d||=ycc}}w)z-Removes all unpicklable entries from hparams.z Attribute 'z' removed from hparams because it cannot be pickled. You can suppress this warning by setting `self.save_hyperparameters(ignore=['z'])`.N)itemsrr )rkv del_attrss rclean_namespacer(sd&}}Ftq!l1oFIF !<<=3e E  AJ Gs A A clsctj|jj}t |j }|dj }dttjdttjjtjjfdttfd}||tjj}||tjj}|||fS)a*Parse key words for standard ``self``, ``*args`` and ``**kwargs``. Examples: >>> class Model: ... def __init__(self, hparams, *my_args, anykw=42, **my_kwargs): ... pass >>> parse_class_init_keys(Model) ('self', 'my_args', 'my_kwargs') rparams param_typercL|D]}|j|k(s|jcSy)N)kindname)r"r#ps r_get_first_if_anyz0parse_class_init_keys.._get_first_if_anyGs- Avv#vv  )inspect signature__init__ parameterslistvaluesr& Parameterr_ParameterKindVAR_POSITIONAL VAR_KEYWORDr str)r init_parameters init_paramsn_selfr(n_argsn_kwargss rparse_class_init_keysr:4s'' 5@@O--/0K ^ FW&&'G22AA7CYCYCeCeef #{G,=,=,L,L MF g.?.?.K.KLH 68 ##r)framec"t|\}}|S)z$For backwards compatibility: #16369.)_get_init_args)r;_ local_argss r get_init_argsr@Vs"5)MAz r)c:tj|\}}}}d|vs|jjdk7rdifS|d}tj|j j }t|\}}}|||fDcgc]}|s| } }g| ddd} |D cic]} | ||  } } |r!| j| j|i| jD cic] \} } | | vs | | } } } |j|d}|| fScc}wcc} wcc} } w)N __class__r,r; frame_args) r* getargvaluesf_codeco_namer+r,r-r:updategetr)r;r> local_varsr r5self_varargs_var kwargs_varn filtered_varsexclude_argnamesrr?rself_args rr=r=\s8!..u5Aq!Z*$ (<(< (JRx [ !C'' 5@@O%:3%?"Hh !)8Z @F1AQFMFKK KWKlK,;rI local_selfr?s rrZrZos("..u5Aq!Z ellEOO 4+E2J j 'Z G5T$ ywWW  yPWXX r))ignorer; given_hparamsargsr\r]ct|dk(rt|ts|dsy|s"tj}|r |j }t|t js td||}nt|rGt|}|Dcic]0}|js|jt||j2}}n-i}ddlm} t!|g| fD]} |j#| |g}nJt|tr|g}n6t|t$t&fr |D cgc]} t| ts| }} t%t)|}|j+D cic] \} } | |vs | | }} } |s|}|rdnd|_nt/|D cgc]\}} t| tr|}}} t|dk(r?||d}|j+D cgc] \} } | |k(s | }} } |r|dnd|_n+|D cic]} t| ts| || }} d|_|j1||j3||j4j+D]2\} } t| t6j8s!t;d| d | d 4t=j>|j4|_ ycc}wcc} wcc} } wcc} }wcc} } wcc} w) zCSee :meth:`~pytorch_lightning.LightningModule.save_hyperparameters`rNz3There is no `frame` available while being required.)HyperparametersMixin)rSkwargsz Attribute z is an instance of `nn.Module` and is already saved during checkpointing. It is recommended to ignore them using `self.save_hyperparameters(ignore=[z])`.)!lenrUr4r* currentframerVrWrXrrrinitr&getattrpytorch_lightning.core.mixinsrarZrGr.tuplesetr _hparams_name enumerate _set_hparamsremove_ignored_hparams_hparamsr Moduler copydeepcopy_hparams_initial)rr\r;r]r^ current_frame init_args obj_fieldsfrar?argrrhpi isx_non_str cand_namess rsave_hyperparametersr|s 4yA~js3DG ,,. !((E eU__ -RSS ! c C[ ;EPaQVVWS!&&11P P F+E2@T?VW )J   Z ( )~ FC  FT5M *!'@#:c3+?#@@ #f+ F"+//"3G$!QqAGIG  (*H(1SVQjc>RqS S { q k!n%B(1(9E1Q"W!EJE1; 1 C 15N#C9M#y~%NBN (C Rv& ""$1 a # QE"^^_]bbfh  ==6CaQAHTFOsH K #K %K;K( K5KK4K$ K2K K%"K%ceZdZdZy)r a0Extended dictionary accessible with dot notation. >>> ad = AttributeDict({'key1': 1, 'key2': 'abc'}) >>> ad.key1 1 >>> ad.update({'my-key': 3.14}) >>> ad.update(new_key=42) >>> ad.key1 = 2 >>> ad "key1": 2 "key2": abc "my-key": 3.14 "new_key": 42 N)__name__ __module__ __qualname____doc__r)rr r sr)r modelzpl.LightningModule attributecg}t||r|j|t|dr)||jvr|j|j|j}||jt|j|r|j|jt|jdr=||jjvr%|j|jj|S)zSpecial attribute finding for Lightning. Gets all of the objects or dicts that holds attribute. Checks for attribute in model namespace, the old hparams namespace/dict, and the datamodule. r)hasattrrYr_trainer datamodule)rrholderstrainers r_lightning_get_all_attr_holdersrsGui uui Y%--%?u}}%nnGw11= 7%%y 1 NN7-- . 7%%y 1i7CUCUC]C]6] NN7--55 6 Nr)cBt||}t|dk(ry|dS)zSpecial attribute finding for Lightning. Gets the object or dict that holds attribute, or None. Checks for attribute in model namespace, the old hparams namespace/dict, and the datamodule, returns the last one that has it. rN)rrc)rrrs r _lightning_get_first_attr_holderr s).eY?G 7|q 2;r)ct||duS)zSpecial hasattr for Lightning. Checks for attribute in model namespace, the old hparams namespace/dict, and the datamodule. N)r)rrs rlightning_hasattrrs ,E9 =T IIr)c|t||}|t|dt|tr||St ||S)a5Special getattr for Lightning. Checks for attribute in model namespace, the old hparams namespace/dict, and the datamodule. Raises: AttributeError: If ``model`` doesn't have ``attribute`` in any of model namespace, the hparams namespace/dict, and the datamodule. _ is neither stored in the model namespace nor the `hparams` namespace/dict, nor the datamodule.)rrrUdictrf)rrholders rlightning_getattrr$sW.eY ?F ~kE E  &$i  69 %%r)valuect||}t|dk(rt|d|D]%}t|tr|||<t |||'y)a]Special setattr for Lightning. Checks for attribute in model namespace and the old hparams namespace/dict. Will also set the attribute on datamodule, if it exists. Raises: AttributeError: If ``model`` doesn't have ``attribute`` in any of model namespace, the hparams namespace/dict, and the datamodule. rrN)rrcrrUrsetattr)rrrrrs rlightning_setattrr:sk.eY?G 7|qkE E  . fd # %F9  FIu - .r))Fr)-rrpr*rrWcollections.abcrr dataclassesrrtypingrrr r torchr pytorch_lightningpllightning_fabric.utilities.datar _AttributeDict%pytorch_lightning.utilities.rank_zeror objectboolrrtyperhr4r:rXrr@r=r.rZr|rrrrrrr)rrsU,  4,00K@f ^  $t$c8C=(3-.O(P$DT#s(^  %// eHSM4S>4Q.R , "  ?? DcN#   49    $sCx. L37'+.2 I7 I7 I7 U8C=#-. /I7 EOO $ I7 DcN+ I7  I7XN$+?CTXY\T]: ,@ S U]^aUb J1JcJdJ&1&c&hsm&,.1.c.#.RV.r)