:j2 ddlZddlZddlmZddlmZmZddlmZm Z ddl m Z m Z m Z ddlmZddlmZde e eee fefd eee ffd Zdeee fd eee ffd Zddee e fd ed ed eee ffdZdeee fd eee ffdZdeee fd eee ffdZde d efdZdeee eeffdeded eee eefffdZy)N) Namespace)MappingMutableMapping)asdict is_dataclass)AnyOptionalUnion)Tensor)_NUMPY_AVAILABLEparamsreturncDt|tr t|}|i}|S)zEnsure parameters are a dict or convert to dict if necessary. Args: params: Target to be converted to a dictionary Returns: params as a dictionary ) isinstancervars)r s v/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/lightning_fabric/utilities/logger.py_convert_paramsrs&&)$f ~ Mcdtdtfd}|jDcic]\}}|||c}}Scc}}w)zSanitize callable params dict, e.g. ``{'a': } -> {'a': 'function_****'}``. Args: params: Dictionary containing the hyperparameters Returns: dictionary with all callables sanitized valrctj|r |jSt|r! |}t|r |jS|S|S#t$rt |ddcYSwxYw)N__name__)inspectisclassrcallable Exceptiongetattr)r_vals r_sanitize_callablez5_sanitize_callable_params.._sanitize_callable:sh ??3 <<  C= 6uD><<'   6sJ55 6sA AA)(A))ritems)r rkeyrs r_sanitize_callable_paramsr"/sA :@ HXS#C#C( ( HH Hs; delimiter parent_keyc i}|jD]\}}|r||zt|zn t|}t|rt|ts t |}nt|t r t|}t|tri|t|||}t|tr=td|Dr+t|D]\}}i|t||d||}|||<|S)aFlatten hierarchical dict, e.g. ``{'a': {'b': 'c'}} -> {'a/b': 'c'}``. Args: params: Dictionary containing the hyperparameters delimiter: Delimiter to express the hierarchy. Defaults to ``'/'``. Returns: Flattened dict. Examples: >>> _flatten_dict({'a': {'b': 'c'}}) {'a/b': 'c'} >>> _flatten_dict({'a': {'b': 123}}) {'a/b': 123} >>> _flatten_dict({5: {'a': 123}}) {'5/a': 123} >>> _flatten_dict({"dl": [{"a": 1, "c": 3}, {"b": 2, "d": 5}], "l": [1, 2, 3, 4]}) {'dl/0/a': 1, 'dl/0/c': 3, 'dl/1/b': 2, 'dl/1/d': 5, 'l': [1, 2, 3, 4]} )r$r#c3<K|]}t|tyw)N)rr).0items r z _flatten_dict..ms(XdD.)I(Xs/) r strrrtyperrrr _flatten_dictlistall enumerate) r r#r$resultkvnew_keyir(s rr-r-Ms* F  15?*y(3q61SV ?:a#6q A 9 %QA a (\\-gQZ"[\F 4 S(XVW(X%X$Q< m4lFlmDyPQRSQTEUaj&kl m F7O  Mrc*|D]}trJddl}t|||j|j|j fr||j ||<t||tttttfvs}t||||<|S)aReturns params with non-primitvies converted to strings for logging. >>> import torch >>> params = {"float": 0.3, ... "int": 1, ... "string": "abc", ... "bool": True, ... "list": [1, 2, 3], ... "namespace": Namespace(foo=3), ... "layer": torch.nn.BatchNorm1d} >>> import pprint >>> pprint.pprint(_sanitize_params(params)) # doctest: +NORMALIZE_WHITESPACE {'bool': True, 'float': 0.3, 'int': 1, 'layer': "", 'list': '[1, 2, 3]', 'namespace': 'Namespace(foo=3)', 'string': 'abc'} rN) r numpyrbool_integerfloatingr(r,boolintfloatr+r )r r2nps r_sanitize_paramsr?us,'  &)bhh BKK%HI"1INN,q q ?4eS&"A AF1IF1I' Mrc|jDcic]\}}|t|s t|n| c}}Scc}}w)z5Convert non-serializable objects in params to string.)r _is_json_serializabler+)r r2r3s r_convert_json_serializablerBs3IO XAA215s1v1 < XX Xs#;valuec |*t|ttttt t fry tj|y#ttf$rYywxYw)z/Test whether a variable can be encoded as json.TF) rr;r<r=r+r.dictjsondumps TypeError OverflowError)rCs rrArAsK } 54eS$*MN 5 } %sAAAmetricsprefix separatorcj|s|S|jDcic]\}}||||c}}Scc}}w)alInsert prefix before each key in a dict, separated by the separator. Args: metrics: Dictionary with metric names as keys and measured quantities as values prefix: Prefix to insert before each key separator: Separates prefix and original key name Returns: Dictionary with prefix and separator inserted before each key )r )rJrKrLr2r3s r _add_prefixrNs< 5<]]_ ETQvhyk! %q ( EE Es/)r*)rrFargparsercollections.abcrr dataclassesrrtypingrr r torchr "lightning_fabric.utilities.importsr rEr+rr"r-r?rBr;rAr=rNrrrWsc 3,''?HU4S>9+D%EF4PSUXPX>(Id38nIc3hI<%.c2%s%VY%cghkmphpcq%PT#s(^S#XBYtCH~Y$sCx.Y   F S% .. /F9<FILF S% & &'Fr