a ij @s<dZddlZddlZddlZddddZddddZdS)z Graph utils.N)returncCsLddlm}ddlm}||j|dtjsHt ddt j d<dS)z;Log stderr & `absl.logging` in Colab (filtered by default).r)logging) googlelogTdefaultZPYTHONWARNINGSN) Zabslr colabtoolsr set_verbosityINFOZset_global_capturesys warnoptionswarnings simplefilterosenviron)rrrV/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/ecolab/patch_utils.py set_verboses     rcstddlm}ddlmddlmddl}|d|dfdd }t|d drpt|d }||j_dS) aFix `graphviz` display on Colab. By default, graphviz object raises an error when displayed on Colab: ``` ExecutableNotFound: failed to execute ['dot', '-Tsvg'], make sure the Graphviz executables are on your systems' PATH ``` Calling this function fix the behavior. r)proto)publish)stubbyNzgraphviz_server.RenderRequestzgraphviz_server.Graphcs4}|j|_dd|d}|jjdS)Nzblade:graphviz-serverzRenderServer.Render)graph)sourcedotCallhtmlZrendered_graphZrendered_bytes)selfrresponseZgraph_proto_clsrZrequest_proto_clsrrr_ipython_display_Bsz)patch_graphviz.._ipython_display_files) rrrrgraphvizZ GetProtoClassgetattrFiler)rrrrrrrpatch_graphviz)s       r")__doc__r r r rr"rrrrs