o
    2h3                     @  sp   U d Z ddlmZ ddlmZ er!ddlmZ ddlmZm	Z	m
Z
 eh dZded< dddZdddZ[dS )z(Integrations with third-party libraries.    )annotations)TYPE_CHECKING)
ModuleType)jaxnumpytorch>   r   r   r   zfrozenset[str]
SUBMODULESreturn	list[str]c                   C  s   g t tdS )Nr   )sortedr    r   r   W/var/www/html/chatgem/venv/lib/python3.10/site-packages/optree/integrations/__init__.py__dir__   s   r   namestrr   c                C  sZ   | t v r#dd l}dd l}|jt }|t d|  }t|| | |S tdtd| )Nr   .zmodule z has no attribute )r   	importlibsysmodules__name__import_modulesetattrAttributeError)r   r   r   module	submoduler   r   r   __getattr__#   s   
r   N)r	   r
   )r   r   r	   r   )__doc__
__future__r   typingr   typesr   optree.integrationsr   r   r   	frozensetr   __annotations__r   r   r   r   r   r   <module>   s   

