U
    /eD                     @  s   d dl mZ d dlZd dlmZmZ eejdd Zeejdd Z	eej
ejfdd	 Zeej
ejfd
d ZdS )    )annotationsN)dask_deserializedask_serializec                 C  s    | j dkrtdd| jig fS )Nrz'Can only serialize read-only h5py filesfilename)mode
ValueErrorr   )f r
   =/tmp/pip-unpacked-wheel-g426oqom/distributed/protocol/h5py.pyserialize_h5py_file   s    
r   c                 C  s   dd l }|j| d ddS )Nr   r   r   )r   )h5pyFile)headerframesr   r
   r
   r   deserialize_h5py_file   s    r   c                 C  s    t | j\}}| j|d< |g fS Nname)r   filer   )xr   _r
   r
   r   serialize_h5py_dataset   s    
r   c                 C  s   t | |}|| d  S r   )r   )r   r   r   r
   r
   r   deserialize_h5py_dataset   s    
r   )
__future__r   r   Zdistributed.protocol.serializer   r   registerr   r   r   GroupZDatasetr   r   r
   r
   r
   r   <module>   s   




