o nio @sddlZddlmZmZmZmZmZddlmZddl m Z m Z edee efZ dZGdddZd ejd ee fd d Zd ddee efdee ded eee effddZdS)N)AnyDictNewTypeOptionalSequence)wandb_internal_pb2) proto_util telemetryBackendConfigDict_wandbc @seZdZdZddeeeefddfddZdeeeffddZ d e j ddfd d Z d eeefddfd dZ d eeefdeeddfddZdejdeededeeeefdef ddZdeededdfddZdeeddfddZdS) ConfigStatezThe configuration of a run.NtreereturncCs|pi|_dSN)_tree)selfr r^/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/internal/sender_config.py__init__s zConfigState.__init__cCsdd|jDS)z+Returns the config settings minus "_wandb".cSsi|] \}}|tkr||qSr)_WANDB_INTERNAL_KEY).0kvrrr sz3ConfigState.non_internal_config..)ritems)rrrrnon_internal_configszConfigState.non_internal_config config_recordcCsB|jD]}|t|t|jq|jD] }|t|qdS)z#Applies update and remove commands.N)update_update_at_path _key_pathjsonloads value_jsonremove_delete_at_path)rr config_itemrrrupdate_from_protos   zConfigState.update_from_protoold_config_treecCs0||g||tdg||tdgdS)z2Merges the config from a run that's being resumed. visualizevizN)_add_unset_keys_from_subtreer)rr'rrrmerge_resumed_config-s z ConfigState.merge_resumed_configpathcCsXt||dd}|s dSt|j|dd}|dusJ|D] \}}||vr)|||<qdS)z8Uses the given subtree for keys that aren't already set.FcreateNT)_subtreerr)rr'r, old_subtree new_subtreekeyvaluerrrr*>s z(ConfigState._add_unset_keys_from_subtreetelemetry_record frameworkstart_time_millismetric_pbdictsc Cs|j}|ti}|j}|r||d<|j}|r||d<|r#||d<|j} | r,| |d<|jj|d<|jj |d<||d<t ||d<|rI||d <t d d |j DS) aReturns a dictionary representation expected by the backend. The backend expects the configuration in a specific format, and the config is also used to store additional metadata about the run. Args: telemetry_record: Telemetry information to insert. framework: The detected framework used in the run (e.g. TensorFlow). start_time_millis: The run's start time in Unix milliseconds. metric_pbdicts: List of dict representations of metric protobuffers. python_version cli_versionr5huggingface_versionis_jupyter_runis_kaggle_kernel start_timetmcSsi|] \}}|d|dqS)N)descr3r)rr2r3rrrrs z/ConfigState.to_backend_dict..)rcopy setdefaultrr8r9r:envjupyterkagglerproto_encode_to_dictr r) rr4r5r6r7 backend_dictwandb_internal py_versionr9r:rrrto_backend_dictOs0     zConfigState.to_backend_dictkey_pathr3cCs4t|j|dddd}|dusJ|||d<dS)z.Sets the value at the path in the config tree.NTr-r/r)rrKr3subtreerrrrs zConfigState._update_at_pathcCs.t|j|dddd}|r||d=dSdS)z3Removes the subtree at the path in the config tree.NrLFr-rM)rrKrNrrrr$szConfigState._delete_at_pathr)__name__ __module__ __qualname____doc__rrstrrrrr ConfigRecordr&r+rr*r TelemetryRecordintr rJrr$rrrrr sP      B  r r%rcCs"|jr|jS|jr |jgStd)z3Returns the key path referenced by the config item.z8Invalid ConfigItem: either key or nested_key must be set) nested_keyr2AssertionError)r%rrrrsrFr-r rKr.cCs8|D]}||}|s|ri}|||<ndS|}q|S)z$Returns a subtree at the given path.N)get)r rKr.r2rNrrrr/s  r/)r typingrrrrr wandb.protor wandb.sdk.librr rSr rr ConfigItemrboolr/rrrrs(