o oi@sddlZddlZddlZddlZddlmZddlmZmZddl m Z m Z e e ZGdddZeddd Zed dd Zed dd Zed Zed ZedZedZedZedZedZedZdS)N) ModuleType)ListOptional)InstallationMode kornia_configc@sveZdZUdZdZeed<ddededdfdd Zdeddfd d Z dd d Z dede fddZ de efddZdS) LazyLoaderaA class that implements lazy loading for Python modules. This class defers the import of a module until an attribute of the module is accessed. It helps in reducing the initial load time and memory usage of a script, especially when dealing with large or optional dependencies that might not be used in every execution. Attributes: module_name: The name of the module to be lazily loaded. module: The actual module object, initialized to None and loaded upon first access. F auto_install module_namedev_dependencyreturnNcCs||_d|_||_dS)aInitialize the LazyLoader with the name of the module. Args: module_name: The name of the module to be lazily loaded. dev_dependency: If the dependency is required in the dev environment. If True, the module will be loaded in the dev environment. If False, the module will not be loaded in the dev environment. N)r moduler )selfr r rH/home/ubuntu/.local/lib/python3.10/site-packages/kornia/core/external.py__init__-s  zLazyLoader.__init__cCs4td|dtjtjdddd|gddddS) Nz Installing `z` ...z-mpipinstallz-UF)shellcheck)loggerinfo subprocessrunsys executable)r r rrr_install_package;s"zLazyLoader._install_packagec Cs|js/dtjvrtd|jddSztr$td|jdWdSWn ty.Ynw|jdurz t |j|_WdSt y}zt j jtjksP|jrW||jn}t j jtjkrd}td|jd}|r|d ksw|d kr||jt |j|_d }n9|d ks|d krd|_||jt |j|_d }n|dks|dkrt d|jd|td}|sknt j jtjkrt d|jd|t |j|_WYd}~dSd}~wwdS)aLoad the module if it hasn't been loaded yet. This method is called internally when an attribute of the module is accessed for the first time. It attempts to import the module and raises an ImportError with a custom message if the module is not installed. z--doctest-modulesz'Doctest detected, skipping loading of ''Nz&Sphinx detected, skipping loading of 'TzOptional dependency 'z' is not installed. You may silent this prompt by `kornia_config.lazyloader.installation_mode = 'auto'`. Do you wish to install the dependency? [Y]es, [N]o, [A]ll.yyesFaallnnoz@' is not installed. Please install it to use this functionality.z-Invalid input. Please enter 'Y', 'N', or 'A'.)r rargvrrr __sphinx_build__ NameErrorr importlib import_module ImportErrorr lazyloaderinstallation_moderAUTOrrASKinputlowerRAISE)r eto_ask if_installrrr_load?sh        zLazyLoader._loaditemcCs|t|j|S)aLoad the module (if not already loaded) and returns the requested attribute. This method is called when an attribute of the LazyLoader instance is accessed. It ensures that the module is loaded and then returns the requested attribute. Args: item: The name of the attribute to be accessed. Returns: The requested attribute of the loaded module. )r3getattrr )r r4rrr __getattr__vs zLazyLoader.__getattr__cCs|t|jS)a|Load the module (if not already loaded) and returns the list of attributes of the module. This method is called when the built-in dir() function is used on the LazyLoader instance. It ensures that the module is loaded and then returns the list of attributes of the module. Returns: list: The list of attributes of the loaded module. )r3dirr )r rrr__dir__s zLazyLoader.__dir__)F)r N)__name__ __module__ __qualname____doc__rbool__annotations__strrrr3objectr6rr8rrrrrs   7rnumpyT)r z PIL.Imageonnx diffusers transformers onnxruntimeboxmotsegmentation_models_pytorchbasicsrrequestsivy)r&loggingrrtypesrtypingrr kornia.configrr getLoggerr9rrrAPILImagerBrCrDrErFrGrHrIrJrrrrs(   z