a ij! @sdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z ej GdddZ ddddd d Zejd d d dZdddddddZdS)zLazy import utils.) annotationsN)AnyIteratorc@sReZdZUdZded<ejedZded<e j ddd d Z dd d d dZ dS) LazyModulez'Module loaded lazily during first call.str module_name)default_factoryzdict[str, LazyModule] _submodulesztypes.ModuleTypereturncCs t|jSN) importlib import_moduler)selfrZ/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/epy/lazy_imports_utils.py_module+szLazyModule._moduler)namer cCs$||jvr|j|St|j|SdSr )r getattrr)rrrrr __getattr__/s  zLazyModule.__getattr__N) __name__ __module__ __qualname____doc____annotations__ dataclassesfielddictr functoolscached_propertyrrrrrrr$s rr)modulerr cCs$t|jd|d}||j|<|S)N.r)rrr )r rZ child_modulerrr_register_submodule9s  r#zIterator[None]r ccs*tj}ztt_dVW|t_n|t_0dS)aContext Manager which lazy loads packages. Their import is not executed immediately, but is postponed to the first call of one of their attributes. Limitation: - You can only lazy load modules (`from x import y` will not work if `y` is a constant or a function or a class). Usage: ```python with epy.lazy_imports(): import tensorflow as tf ``` Yields: None N)builtins __import__ _lazy_import)Zoriginal_importrrr lazy_importsAs r'rztuple[str, ...]int)rfromlistlevelc Csl~~|rtd|d|d^}}t|d}|}|D]}t||}q8|rd|D]} t|| qP|S|SdS)zMock of `builtins.__import__`.z*Relative import statements not supported (z).r!r"N) ValueErrorsplitrr#) rglobals_Zlocals_r)r*Z root_namepartsrootchildflrrrr&as   r&)NNrr)r __future__rr$ contextlibrrr typestypingrr dataclassrr#contextmanagerr'r&rrrrs$ !