U
    ÷Õ/er  ć                
   @  sŲ  d dl mZ d dlmZ d dlmZ d dlmZmZ d dl	m
Z
mZmZmZmZ d dlmZmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z  e !d”e !d”d	d
 Z"e !d”e !d”dd Z#e !d”e !d”dd Z$e !d”e !d”dd Z%e !d”e !d”dd Z&e !d”e !d”dd Z'e !d”e !d”dd Z(e !d”e !d”e !d”e !d”dd  Z)e !d!”e !d!”e !d!”e !d!”e !d"”e !d"”e !d"”e !d"”d#d$ Z*e !d%”e !d%”e !d%”e !d%”d&d' Z+e !d(”e !d(”e !d(”e !d(”d)d* Z,e !d+”e !d+”e !d+”e !d+”e !d,”e !d,”e !d,”e !d,”d-d. Z-e !d/”e !d/”e !d/”e !d/”d0d1 Z.d2S )3é    )Śannotations)Śsuppress)Śpartial)ŚcompressionsŚdefault_compression)Ś
decompressŚdumpsŚloadsŚmaybe_compressŚmsgpack)Ścuda_deserializeŚcuda_serialize)Ś	SerializeŚ
SerializedŚdask_deserializeŚdask_serializeŚdeserializeŚdeserialize_bytesŚnested_deserializeŚregister_genericŚregister_serializationŚregister_serialization_familyŚ	serializeŚserialize_bytelistŚserialize_bytesŚto_serializeŚnumpyc                  C  s   ddl m}  d S )Nr   ©r   )Śdistributed.protocolr   r   © r   śA/tmp/pip-unpacked-wheel-g426oqom/distributed/protocol/__init__.pyŚ_register_numpy   s    r!   Śscipyc                  C  s   ddl m}  d S )Nr   ©r"   )r   r"   r#   r   r   r    Ś_register_scipy!   s    r$   Śh5pyc                  C  s   ddl m}  d S )Nr   ©r%   )r   r%   r&   r   r   r    Ś_register_h5py'   s    r'   ZnetCDF4c                  C  s   ddl m}  d S )Nr   ©Śnetcdf4)r   r)   r(   r   r   r    Ś_register_netcdf4-   s    r*   Śkerasc                  C  s   ddl m}  d S )Nr   ©r+   )r   r+   r,   r   r   r    Ś_register_keras3   s    r-   Śsparsec                  C  s   ddl m}  d S )Nr   ©r.   )r   r.   r/   r   r   r    Ś_register_sparse9   s    r0   Zpyarrowc                  C  s   ddl m}  d S )Nr   ©Śarrow)r   r2   r1   r   r   r    Ś_register_arrow?   s    r3   ŚtorchZtorchvisionc                  C  s   ddl m}  d S )Nr   ©r4   )r   r4   r5   r   r   r    Ś_register_torchE   s    r6   ŚcupyZcupyxc                  C  s   ddl m}  d S )Nr   ©r7   )r   r7   r8   r   r   r    Ś_register_cupyM   s    	r9   Śnumbac                  C  s   ddl m}  d S )Nr   ©r:   )r   r:   r;   r   r   r    Ś_register_numbaY   s    r<   Śrmmc                  C  s   ddl m}  d S )Nr   ©r=   )r   r=   r>   r   r   r    Ś_register_rmma   s    r?   ZcudfZ	dask_cudfc                  C  s   ddl m}  d S ©Nr   ©r   )Z	cudf.commr   rA   r   r   r    Ś_register_cudfi   s    	rB   Zcumlc               	   C  s$   t t ddlm}  W 5 Q R X d S r@   )r   ŚImportErrorZ	cuml.commr   rA   r   r   r    Ś_register_cumlu   s    
rD   N)/Ś
__future__r   Ś
contextlibr   Ś	functoolsr   Z distributed.protocol.compressionr   r   Zdistributed.protocol.corer   r   r	   r
   r   Zdistributed.protocol.cudar   r   Zdistributed.protocol.serializer   r   r   r   r   r   r   r   r   r   r   r   r   r   Zregister_lazyr!   r$   r'   r*   r-   r0   r3   r6   r9   r<   r?   rB   rD   r   r   r   r    Ś<module>   s   @