a
ijB @ s d Z ddlmZ ddlZddlZddlZddlZddlZddl Z ddl m
Z
m Z m Z ddl
Z
ddlZe jreeB eB eB dB Zeeedf B ed B Ze dZejddd d
d d d
ddZddd ddddZdddddZ dS )z Colab utils. ) annotationsN)IteratorIterableTypeVarJson_T F)expandedstrboolzIterator[None])namer returnc c s ddl }| }|j|gd}|d| |r6d|_nd|_tj| |J z
dV W n* ty } z|} W Y d}~nd}~0 0 d}W d n1 s0 Y |dur|dS )zCapture all outputs and display it in a collapsible block.
Args:
name: Name of the collapsible section.
expanded: If `True`, the section is expanded by default.
Yields:
None
r N)children)
ipywidgetsOutputZ Accordion set_titleZselected_indexIPythondisplay Exception)r r r outZ accordioneexc r V/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/ecolab/colab_utils.pycollapse% s"
"r None)valuer r
c C s t j}d}tt| }|r.d| dnd}d| d| d| d| d | d
| d | d | d
| d}tjtj | dS )a Display the Json `dict` / `list` interactivelly (with collapsible elems).
Examples:
```python
ecolab.json({'a': [1, 2, 3], 'b': {'x': True, 'y': False}})
```
The dict keys and list indices can be filtered from the display field using
regex (e.g. `a.[0-9]` in the above example).
Args:
value: Json `dict` or `list` to inspect.
expanded: Whether the elements start as expanded or as collapsed.
a/
json-viewer {
padding:1px 1.5em 1px 1.5em;
--background-color: #f7f7f7;
--property-color: #087db2;
--string-color: #a31515;
--number-color: #008000;
--boolean-color: #0000ff;
--null-color: #af00db;
--preview-color: #888888;
}
json-viewer::part(key) {
margin-right: 0.5em;
}
html[theme=dark] json-viewer {
--background-color: #2c2c2c;
--property-color: #6fb3d2;
--string-color: #ce9178;
--number-color: #b5cea8;
--boolean-color: #569cd6;
--null-color: #c586c0;
--preview-color: #888888;
}
.ecolab-json button {
background-color: var(--colab-highlighted-surface-color);
color: var(--colab-primary-text-color);
border-width: 0;
}
.ecolab-json input {
border-color: var(--colab-highlighted-surface-color);
}
Zviewerz.expandAll();
r z
z
N)
uuiduuid1hexhtmlescapejson_stddumpsr r HTML)r r Zid_Z css_contentZ html_contentZ expand_liner r r jsonH s0
!
r% zIterable[_T] | Iterator[_T]z Iterator[_T])innerr
c
# sx t fdd}ttj|}z@| D ](}|V r( W ttj| dS q(W ttj| nttj| 0 dS )a Catch KeyboardInterrupts to end the loop without raising an Exception.
While this iterator is running, the first SIGINT (e.g. from interrupting
the colab runtime, or pressing Ctrl+C ) will not raise an exception but
instead end the loop after the current iteration.
A second SIGINT will raise a KeyboardInterrupt Exception as usual.
Examples:
```python
for i in interruptible(range(10)):
print(i)
time.sleep(3)
```
Args:
inner: arbitrary iterable or iterator
Yields:
elements from inner
c s r t d S )N)is_setKeyboardInterruptset)Z
unused_signumZ unused_frame interruptedr r handler s z interruptible..handlerN) threadingEventsignalSIGINTr' )r& r, Zprevious_handlerir r* r
interruptible s r2 )r )F) __doc__
__future__r
contextlibr r% r" r/ r- typingr r r r IPython.displayr
TYPE_CHECKINGr
floatintr Z JsonValuedictlistr r contextmanagerr r2 r r r r s$ "P