o -wi @sdddlmZddlmZddlmZddlZeGdddZeGdddZ  ddddZ dS)) annotations) dataclass)AnyNc@steZdZUded<ded<ded<dZded<dZded <d Zd ed <edddZedddZ edddZ dS)CustomChartSpecstr spec_namedict[str, Any]fields string_fieldskeyVega2 panel_typeFbool split_tablereturncCs0|jstd|jrd|jdS|jdS)Nz)Key for the custom chart spec is not set.zCustom Chart Tables/_table)r wandbErrorrselfrT/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/plot/custom_chart.py table_keys   zCustomChartSpec.table_keycCsd|j|j|j|jddidddddgdgddgdd gdd d |jdggd gd gid dS)NnametableWithLeafColNames queryFieldsrunSetsz ${runSets})rvalueid)rr _defaultColorIndex summaryTabletableKey)rargsr ) panelDefId fieldSettingsstringSettings transform userQuery)r panel_config)rrr r rrrrr config_values0 zCustomChartSpec.config_valuetuple[str, str, str]cCs dd|jfS)N_wandb visualize)r rrrr config_key= zCustomChartSpec.config_keyN)rr)rr)rr+) __name__ __module__ __qualname____annotations__r rrpropertyrr*r.rrrrr s      "rc@s(eZdZUded<ded<d ddZd S) CustomChart wandb.Tabletablerspecr rcCs ||j_dS)z?Sets the key for the spec and updates dependent configurations.N)r8r )rr rrrset_keyGr/zCustomChart.set_keyN)r r)r0r1r2r3r9rrrrr5Bs r5Fvega_spec_namer data_tabler6r rr dict[str, Any] | NonerrrcCs>t|tjstdt|jt|t|||pi|ddS)axCreates a custom charts using a Vega-Lite specification and a `wandb.Table`. This function creates a custom chart based on a Vega-Lite specification and a data table represented by a `wandb.Table` object. The specification needs to be predefined and stored in the W&B backend. The function returns a custom chart object that can be logged to W&B using `wandb.log()`. Args: vega_spec_name (str): The name or identifier of the Vega-Lite spec that defines the visualization structure. data_table (wandb.Table): A `wandb.Table` object containing the data to be visualized. fields (dict[str, Any]): A mapping between the fields in the Vega-Lite spec and the corresponding columns in the data table to be visualized. string_fields (dict[str, Any] | None): A dictionary for providing values for any string constants required by the custom visualization. split_table (bool): Whether the table should be split into a separate section in the W&B UI. If `True`, the table will be displayed in a section named "Custom Chart Tables". Default is `False`. Returns: CustomChart: A custom chart object that can be logged to W&B. To log the chart, pass it to `wandb.log()`. Raises: wandb.Error: If `data_table` is not a `wandb.Table` object. zs  8