o xi@sddlmZddlZddlmZddlmZmZddlm Z ddl m Z m Z ede eefZ dZGd d d ZdddZdddddZdS)) annotationsN)Sequence)AnyNewType)wandb_internal_pb2) proto_util telemetryBackendConfigDict_wandbc@sbeZdZdZd+d,ddZd-d d Zd.ddZd/ddZd0ddZd1d"d#Z d2d'd(Z d3d)d*Z dS)4 ConfigStatezThe configuration of a run.Ntreedict[str, Any] | NonereturnNonecCs|pi|_dSN)_tree)selfr rT/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/internal/sender_config.py__init__s zConfigState.__init__dict[str, Any]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_recordwandb_internal_pb2.ConfigRecordcCsB|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_proto"s   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_config0s z ConfigState.merge_resumed_configpath Sequence[str]cCsXt||dd}|s dSt|j|dd}|dusJ|D] \}}||vr)|||<qdS)z8Uses the given subtree for keys that aren't already set.FcreateNT)_subtreerr)rr*r0 old_subtree new_subtreer.valuerrrr->s z(ConfigState._add_unset_keys_from_subtreetelemetry_recordtelemetry.TelemetryRecord framework str | Nonestart_time_millisintmetric_pbdictsSequence[dict[int, Any]]environment_record$wandb_internal_pb2.EnvironmentRecordr c 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_versionr:huggingface_versionis_jupyter_runis_kaggle_kernel start_timetmecSsi|] \}}|d|dqS)N)descr7r)rr.r7rrrrs z/ConfigState.to_backend_dict..)rcopy setdefaultrrBrCrDenvjupyterkagglerproto_encode_to_dict writer_idmessage_to_dictr r) rr8r:r<r>r@ backend_dictwandb_internal py_versionrCrDrRenvironment_dictrrrto_backend_dictOs8       zConfigState.to_backend_dictkey_pathr7rcCs4t|j|dddd}|dusJ|||d<dS)z.Sets the value at the path in the config tree.NTr2r4r)rrYr7subtreerrrr!s zConfigState._update_at_pathcCs.t|j|dddd}|r||d=dSdS)z3Removes the subtree at the path in the config tree.NrZFr2r[)rrYr\rrrr'szConfigState._delete_at_pathr)r r rr)rr)rrrr)r*rrr)r*rr0r1rr) r8r9r:r;r<r=r>r?r@rArr )rYr1r7rrr)rYr1rr) __name__ __module__ __qualname____doc__rrr)r/r-rXr!r'rrrrr s      K r r(wandb_internal_pb2.ConfigItemrr1cCs"|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(rrrr"sr"Fr2r rrYr3boolr cCs8|D]}||}|s|ri}|||<ndS|}q|S)z$Returns a subtree at the given path.N)get)r rYr3r.r\rrrr4s  r4)r(rarr1)r rrYr1r3rdrr ) __future__rr#collections.abcrtypingrr wandb.protor wandb.sdk.librrdictstrr rr r"r4rrrrs