a na!@sNddlmZddlZeeZddlmZddlm Z dZ ddZ dd Z dS) ) annotationsN)StaticLayoutProvider) GraphRenderer from_networkxc  st}ddt|jddD}tt|}|D].fdd|jddD}t|}||<q0d|vr|ddlm}|d t||d<t}d d|jddD}tt|}|D].fd d|jddD}t|}||<qd |vsd |vrddlm}|ddd|D|d <dd|D|d <t} || j j _ || j j _ t |rp||fi|} n<|} | j j j d} t| t|krddlm}|dt| d| _| S)a* Generate a ``GraphRenderer`` from a ``networkx.Graph`` object and networkx layout function. Any keyword arguments will be passed to the layout function. Only two dimensional layouts are supported. Args: graph (networkx.Graph) : a networkx graph to render layout_function (function or dict) : a networkx layout function or mapping of node keys to positions. The position is a two element sequence containing the x and y coordinate. Returns: instance (GraphRenderer) .. note:: Node and edge attributes may be lists or tuples. However, a given attribute must either have *all* lists or tuple values, or *all* scalar values, for nodes or edges it is defined on. .. warning:: Node attributes labeled 'index' and edge attributes labeled 'start' or 'end' are ignored. If you want to convert these attributes, please re-label them to other names. Raises: ValueError cSs"g|]}|dD]}|qqSkeys).0Znodeattr_keyr 3lib/python3.9/site-packages/bokeh/plotting/graph.py Esz!from_networkx..T)datacs(g|] \}}|vr |ndqSNr )r _Z node_attrr r rrJsindexr)warnzConverting node attributes labeled 'index' are skipped. If you want to convert these attributes, please re-label with other names.cSs"g|]}|dD]}|qqS)rr )r Zedger r r rrZscs*g|]"\}}}|vr"|ndqSrr )r rZ edge_attrrr rr_sstartendzConverting edge attributes labeled 'start' or 'end' are skipped. If you want to convert these attributes, please re-label them with other names.cSsg|] }|dqS)rr r xr r rrkcSsg|] }|dqSrr rr r rrlrznNode keys in 'layout_function' don't match node keys in the graph. These nodes may not be displayed correctly.) graph_layout)dictlistZnodesset_handle_sublistswarningsrZedgesrZ node_rendererZ data_sourcerZ edge_renderercallabler rZlayout_provider) ZgraphZlayout_functionkwargsZ node_dictZnode_attr_keysvaluesrZ edge_dictZedge_attr_keysZgraph_rendererrZ node_keysr rrr$sN                cCs>tdd|Dr:tdd|Ds,tddd|DS|S)Ncss|]}t|ttfVqdSr isinstancertuplerr r r rz#_handle_sublists..css$|]}|durt|ttfVqdSrr$rr r rr'rz;Can't mix scalar and non-scalar values for graph attributescSs g|]}|durgnt|qSr)rrr r rrrz$_handle_sublists..)anyall ValueError)r#r r rrs r) Z __future__rZloggingZ getLogger__name__logZ models.graphsrZmodels.renderersr__all__rrr r r r s    e