o -wi @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 eeefdeeddfddZdejdeededeeeefde jdef ddZdeededdfddZdeeddfddZdS) ConfigStatezThe configuration of a run.NtreereturncCs|pi|_dSN)_tree)selfr r]/home/ubuntu/sommelier/.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_treecCs*||gdD] }||t|gqdS)z2Merges the config from a run that's being resumed.)viz visualizezmask/class_labelsN)_add_unset_keys_from_subtreer)rr'keyrrrmerge_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_subtreer+valuerrrr*;s z(ConfigState._add_unset_keys_from_subtreetelemetry_record frameworkstart_time_millismetric_pbdictsenvironment_recordc Cs|j}|ti}|j}|r||d<|j} | r| |d<|r#||d<|j} | r,| |d<|jj|d<|jj |d<||d<t ||d<|rI||d <|j } | rYt |} | | i|d <td d |jDS) 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_timetmecSsi|] \}}|d|dqS)N)descr3r)rr+r3rrrrs z/ConfigState.to_backend_dict..)rcopy setdefaultrr9r:r;envjupyterkagglerproto_encode_to_dict writer_idmessage_to_dictr r) rr4r5r6r7r8 backend_dictwandb_internal py_versionr:r;rIenvironment_dictrrrto_backend_dictLs8       zConfigState.to_backend_dictkey_pathr3cCs4t|j|dddd}|dusJ|||d<dS)z.Sets the value at the path in the config tree.NTr.r0r)rrPr3subtreerrrrs zConfigState._update_at_pathcCs.t|j|dddd}|r||d=dSdS)z3Removes the subtree at the path in the config tree.NrQFr.rR)rrPrSrrrr$szConfigState._delete_at_pathr)__name__ __module__ __qualname____doc__rrstrrrrr ConfigRecordr&r,rr*r TelemetryRecordintEnvironmentRecordr rOrr$rrrrr sT      K  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_keyr+AssertionError)r%rrrrsrFr.r rPr/cCs8|D]}||}|s|ri}|||<ndS|}q|S)z$Returns a subtree at the given path.N)get)r rPr/r+rSrrrr0s  r0)r typingrrrrr wandb.protor wandb.sdk.librr rXr rr ConfigItemrboolr0rrrrs(