o
    㥵i                  	   @   sl   d dl Z d dlZd dlmZ d dlmZmZmZ d dlZ	eG dd dZ
	dd	ed
ededefddZdS )    N)	dataclass)LiteralOptionalTuplec                   @   s<   e Zd ZU ed ed< eeeej	f  ed< ee
 ed< dS )InferenceResult)headersegmenterrorfinalcodeaudior	   N)__name__
__module____qualname__r   __annotations__r   r   intnpndarray	Exception r   r   V/home/ubuntu/.local/lib/python3.10/site-packages/fish_speech/inference_engine/utils.pyr   	   s   
 r   D        sample_rate	bit_depthchannelsreturnc                 C   sj   t  }t|d}|| ||d  ||  W d    n1 s&w   Y  | }|  |S )Nwb   )	ioBytesIOwaveopensetnchannelssetsampwidthsetframerategetvalueclose)r   r   r   bufferwav_filewav_header_bytesr   r   r   wav_chunk_header   s   
r,   )r   r   r   )r    r"   dataclassesr   typingr   r   r   numpyr   r   r   bytesr,   r   r   r   r   <module>   s"    