o
    ÉÙ¾iÜ  ã                   @   sD   d Z ddlmZ ddlmZ ddlmZ dd„ Zd	d
„ Zdd„ Z	dS )zDECORD MXNet bridgeé    )Úabsolute_importé   )Ú_from_dlpacké   ©Ú
try_importc                  C   s   d} t d| ƒS )zqTry import mxnet at runtime.

    Returns
    -------
    mxnet module if found. Raise ImportError otherwise
    zsmxnet is required, you can install by pip.
         CPU: `pip install mxnet-mkl`, GPU: `pip install mxnet-cu100mkl`Úmxnetr   )Úmsg© r
   úG/home/ubuntu/.local/lib/python3.10/site-packages/decord/bridge/mxnet.pyÚtry_import_mxnet   s   
r   c                 C   s   t ƒ }|j |  ¡ ¡S )zfrom decord to mxnet, no copy)r   ÚndÚfrom_dlpackÚ	to_dlpack)Ú
decord_arrÚmxr
   r
   r   Úto_mxnet   s   r   c                 C   s   t |  ¡ ƒS )zfrom mxnet to decord, no copy)r   Úto_dlpack_for_read)Ú	mxnet_arrr
   r
   r   Ú
from_mxnet   s   r   N)
Ú__doc__Ú
__future__r   Ú_ffi._ctypes.ndarrayr   Úutilsr   r   r   r   r
   r
   r
   r   Ú<module>   s    