
    0Fie8                    F   d dl mZ d dlZd dlmZmZ ej        dk     r ed           ej        ej	                  d             Z
 ej        ej	                  d             Z ej        ej                  d             Z ej        ej                  d	             ZdS )
    )annotationsN)dask_deserializedask_serializez0.10zLNeed pyarrow >= 0.10 . See https://arrow.apache.org/docs/python/install.htmlc                    t          j                    }t          j        || j                  }|                    |            |                                 |                                }i }|g}||fS N)pyarrowBufferOutputStreamRecordBatchStreamWriterschemawrite_batchclosegetvalue)batchsinkwriterbufheaderframess         :lib/python3.11/site-packages/distributed/protocol/arrow.pyserialize_batchr      sh    %''D,T5<@@F
u
LLNNN
--//CFUF6>    c                    |d         }t          j        t          j        |                    }|                                S Nr   )r   RecordBatchStreamReaderBufferReaderread_next_batchr   r   blobreaders       r   deserialize_batchr       s8    !9D,W-A$-G-GHHF!!###r   c                    t          j                    }t          j        || j                  }|                    |            |                                 |                                }i }|g}||fS r   )r   r	   r
   r   write_tabler   r   )tblr   r   r   r   r   s         r   serialize_tabler$   !   sh    %''D,T3:>>F
s
LLNNN
--//CFUF6>r   c                    |d         }t          j        t          j        |                    }|                                S r   )r   r   r   read_allr   s       r   deserialize_tabler'   -   s6    !9D,W-A$-G-GHHF??r   )
__future__r   r   distributed.protocol.serializer   r   __version__ImportErrorregisterRecordBatchr   r    Tabler$   r'    r   r   <module>r0      s   " " " " " "  K K K K K K K K

+	@   ,--  .- 7.//$ $ 0/$ ''  (' 7=))  *)  r   