a
    ij                     @  s  U d Z ddlmZ ddlmZ ddddZddd	d
Zee Z	e	j
dd ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl!Z!ddl"Z"ddl#Z#ddl$Z$ddl%Z%ddl&Z&ddl'Z'ddl(Z(ddl)Z)ddl*Z*ddl+Z+ddl,Z,ddl-Z-ddl.Z.ddl/Z/ddl0Z0ddl1Z1ddl2Z2ddl3Z3ddl4Z4ddl5Z5ddl6Z6ddl7Z7ddl8Z8ddl9Z9ddl:Z:ddl;Z;ddl<Z<ddl=Z=ddl>Z>ddl?Z?ddl@Z@ddlAZAddlBmCZC ddlDZDddlEZEddlFZFW d   n1 s@0    Y  e	j
dd ddlGmHZH ddlGmIZI ddlGmJZJ ddlGmKZK ddlGmLZL ddlGmMZM ddlGmNZN ddlGmOZO ddlmPZP ddlQmRZR ddlQmSZS ddlTZUddlVZVddlWZXddlYZYddlZZZddl[Z[ddl[m\Z] ddl^Z^ddl_Z_ddl`maZb ddlcZcddldZdddleZeddlfZfddlfmgZh ddliZiddljZjddlkZkddlkZlddlkmmZn ddloZpddlqZqddlrZsddlgZtddluZuddlvZvddlvmwZx ddlvmyZz ddl{Z|ddl}Z}dd l}m~Z~ ddlZddlZddlZddlZddlZddlm  mgZ ddlZddlZddlZddlZddlZej ej ddlZddlZddlZddlZdd!lmZ dd"lmZ ddlZddlZddlZddlZdd#lmZ dd$lmZ W d   n1 s0    Y  eee	j d%d& d'Zd(ed)< eeZdS )*a  Common lazy imports.

Usage:

```python
from etils.ecolab.lazy_imports import *
```

To get the list of available modules:

```python
lazy_imports.__all__  # List of modules aliases
lazy_imports.LAZY_MODULES  # Mapping <module_alias>: <lazy_module info>
```
    )annotations)
lazy_utilsz	list[str])returnc                   C  s   g dS )z`lazy_imports` public API.

  Because `globals()` contains hundreds of symbols, we overwrite `dir(module)`
  to avoid poluting the namespace during auto-completion.

  Returns:
    public symbols
  )__all__LAZY_MODULESprint_current_imports r   r   r   W/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/ecolab/lazy_imports.py__dir__$   s    
r
   Nonec                   C  s   t tt dS )zDisplay the active lazy imports.

  This can be used before publishing a colab. To convert lazy imports
  into explicit imports.

  For convenience, `from etils.ecolab import lazy_imports` is excluded from
  the current imports.
  N)printr   Zcurrent_import_statementsr   r   r   r   r	   r   5   s    	r   T)is_stdN)mockF)array_types)ecolab)edc)enp)epath)epy)etqdm)etree)lazy_imports)app)flags)linen)numpy)pyplot)
checkpoint)export)Image)express)graph_objects)flow)xmc                 C  s   | d j jS )N   )Z_etils_statemodule_name)xr   r   r	   <lambda>       r'   )keyz dict[str, lazy_utils.LazyModule]r   )__doc__
__future__r   Zetils.ecolabr   r
   r   ZLazyImportsBuilderglobalsZ_builderZreplace_importsabcargparseastZasynciobase64builtinscollectionscolorsyscopyconcurrent.futuresZ
concurrent
contextlibZcontextvarscsvdataclassesdatetimedifflibdisenum	functoolsgcgziphtmlinspectio	importlibZIPython	itertoolsjsonloggingmathmultiprocessingospathlibZpdbpicklepprintqueuerandomreshutilstatstring
subprocesssystarfiletextwrap	threadingtimeZtimeittomllib	tracebacktypingtypesuuidZunittestr   warningsweakrefzipfileZetilsr   r   r   r   r   r   r   r   r   Zabslr   r   Zapache_beamZbeamZchexZdataclass_arrayZdcaZeinopsZflaskZflaxr   nnZ	functorchZginZgrain.pythonpythonZgraingraphvizZimageio
ipywidgetsZjaxr   ZjnpZ	jaxtypingZlarkZ
matplotlibZmplr   ZpltZmediapyZmediaZml_collectionsZnetworkxnxnpZoptaxZorbaxr   Zocpr   ZoexZpandaspdZPILr   ZpycolmapZscipyZseabornZsnsZsklearnZ
tensorflowtfZtensorflow.experimental.numpyZexperimentalZtnpZtensorflow_datasetsZtfdsZtorchZ	torchtextZtorchvisionZtqdmautonotebooktreeZ	typeguardZtyping_extensionsZplotlyr    Zpxr!   goZpydanticrequestsZsundsZvisu3dZv3dZxmanager.contribr"   ZxmflowZxmanagerr#   dictsortedZlazy_modulesitemsr   __annotations__r   r   r   r   r	   <module>   s  (,