o xi6@s&UddlmZddlZddlZddlZddlmZmZddlZddl Zddl m Z er5ddl Z ddlmZddiddddad ed <dad ed <d gZejdZejdp]ejdZercejndZd/ddZd0d1ddZd0d2ddZ d0d3d"d#Zd4d%d&Z  d5d6d+d,Zd7d8d-d.ZdS)9) annotationsN) TYPE_CHECKINGAny) telemetry) TBHistorysteprlast_logglobalzdict[str, dict[str, Any]]STEPSzfloat | int | NoneRATE_LIMIT_SECONDSgraphsztensorboard.util.tensor_utilz$tensorboard.compat.proto.summary_pb2z%tensorflow.core.framework.summary_pb2tensorrreturnnp.ndarray | NonecCs.trt|}|jdkrdS|StddS)NobjectzZCan't convert tensor summary, upgrade tensorboard with `pip install tensorboard --upgrade`) tensor_util make_ndarraydtypewandbtermwarn)rresrU/home/ubuntu/.local/lib/python3.10/site-packages/wandb/integration/tensorboard/log.pyr's  rr tagstr namespacecCs|s|S|d|S)N/r)rrrrrnamespaced_tag7s r keycCsttdd||S)a\Convert invalid filesystem characters to _ for use in History keys. Unfortunately this means currently certain image keys will collide silently. We implement this mapping up here in the TensorFlow stuff rather than in the History stuff so that we don't have to store a mapping anywhere from the original keys to the safe ones. z[/\\]_)r resub)r!rrrrhistory_image_key>sr%tf_summary_str_or_pbdict[str, Any] | Nonec sit|dr|j}|jtd<|jd<nEt|tttfr)t }| |n4t|dr[ddt t |D}t |D]\}}|| ||dkrU|d||q=|d}n|}t|drit |jdkrkd Sd4fdd }|jD]}|d}|tvrqv|dkr|jt|j<qv|dkr|jjj}|dks|dkrt|jt|j<qv|dkr|jjdd } || |qv|dkrGt|j} | d urqv| j} g} g} | ddkr| | dd| D]}| |d| |dqn| ddkr| ddg} | dd d} t | dkrFztj| | fdt|j<WqvtyEtjdt|jdtjj dddYqvwqv|d krt|j}|d urWqv|d!d d f!}|d"d d f!}g}t t"t |t |D]}||dks||dkr|||||fqxt#|d#d$d%}tj$|d&d'gd(}t|j}t%d)|d&d'd*d+|d,i|<qv|d-kr|j&j'}||g|qv|d.krKt|j}t |j(j)d/kr@|j(j)d|j(j)d|j(j)d}|j(j)d!|j(j)d!|j(j)d0}t*|j(j+|g|j(j)d d"|gf}z tj|d|<Wqvty?tjd1|d2tjj dddYqvwtjd1|d3ddqvS)5zConvert a Tensorboard Summary to a dictionary. Accepts a tensorflow.summary.Summary, one encoded as a string, or a list of such encoded as strings. summary global_step _timestamp__iter__cSsg|]}tqSr)Summary).0r"rrr Zz&tf_summary_to_dict..rvalueN _img_strs list[bytes]_valuerrNonec szddlm}WntytjdddYdSwt|dkr"dSg}|D]#}|dr;|tjt |ddq&|t| t |q&|j d d }t|d kro|d ro|\}}t|g|dS|t|j <dS) Nr)Imagez`Install pillow if you are logging images with Tensorboard. To install, run `pip install pillow`.FrepeatsGIFgif)formatr)PILr5 ImportErrorrrlen startswithappendVideoioBytesIOopenrrsplitisdigit setdefaultr%extend)r1r3r5images_img_strtag_idxridxrvaluesrr encode_imagesgs.   z)tf_summary_to_dict..encode_images simple_valuerscalarsr rH histogramsr:) np_histogramzNot logging key "z#". Histograms must have fewer than z binsFr6 pr_curvescSs|d|d fS)Nrr:r)xrrrr/z$tf_summary_to_dict..)r!recall precision)datacolumnsz wandb/line/v0)rWytitlez Precision v. RecallimagehistozNot logging key z". Histograms must have fewer than z%. Found a histogram with only 2 bins.)r1r2r3rrr4),hasattrr(rr wall_time isinstancerbytes bytearrayr,ParseFromStringranger= enumerate MergeFromr0 WhichOneof IGNORE_KINDSrOrmetadata plugin_data plugin_namerr string_valshaper?r Histogram ValueErrorr MAX_LENGTHtolistminsortedTable plot_tabler_encoded_image_stringr` bucket_limitlistbucket)r&r summary_pbir(rNr0kindrpimg_strsndarrayrrcountsbinsv pr_curve_datarZrYr[ data_tablenameimg_strrfirstlastrSrrLrtf_summary_to_dictIs                               rr4cCsddiddddadS)z>Internal method for resetting state, called by wandb.finish().rrNrr )r rrrr reset_state srhistoryTBHistory | Noneintkwargsc KsLt|ddi}d}|d|krd}t||}|durdS|dt}tdddur3|tdd<|r?|tddkr?d}|tddkrM|tdd<|d krYtdd|d <d|it|<|rtduso|tddtkr|durtjdur~tjin|i|tdd<|durtjdurtjj|dd dSdS| |dS) aKLogs a tfsummary to wandb. Can accept a tf summary string or parsed event. Will use wandb.run.history unless a history object is passed. Can optionally namespace events. Results are committed when step increases for this namespace. NOTE: This assumes that events being passed in are in chronological order rrFTNr*r r r r))commit) r getrtimerrrun_logadd _row_update) r&rrrr last_steprlog_dict timestamprrrrs>          rcCsVtjdur tdt }d|j_Wdn1swYt|||ddS)NzBYou must call `wandb.init()` before calling `wandb.tensorflow.log`T)rr)rrErrorrcontextfeaturetensorboard_logr)r&rrtelrrrlogXs   r)rrrr)r )rrrrrr)r!rrrrr)r&rrrrr')rr4)Nrr ) r&rrrrrrrrrrr4)rr )r&rrrrrrr4) __future__rrAr#rtypingrrr wandb.util wandb.sdk.librnumpynpwandb.sdk.internal.tb_watcherrr __annotations__rrmutil get_modulerpbr,rr r%rrrrrrrrsF          E E