o
    ni
                     @   s   d dl Z d dlZd dlZd dlmZmZ eeZzJd dl	Z	ej
drUdededefddZded	ededdfd
dZdeeef defddZd	edefddZW dS d dl mZmZmZmZ W dS  eyy   d dl mZmZmZmZ Y dS w )    N)AnyUnion_WANDB_ORJSONobjkwargsreturnc              
   K   s   | dd}z| }|r|j|d< tj| fdtji| }W |S  tyL } zt	d|  |r8||d< t
j| fi |}W Y d}~|S d}~ww )z Wrapper for <json|orjson>.dumps.clsNdefaultoptionzError using orjson.dumps: )popcopyr	   orjsondumpsOPT_NON_STR_KEYSdecode	Exceptionlogger	exceptionjson)r   r   r   _kwargsencodede r   U/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/sdk/lib/json_util.pyr      s.   
	r   fpc              
   K   s   | dd}z| }|r|j|d< tj| fdtji|}|| W dS  tyP } zt	d|  |r;||d< t
j| |fi | W Y d}~dS d}~ww )zWrapper for <json|orjson>.dump.r   Nr	   r
   zError using orjson.dump: )r   r   r	   r   r   r   writer   r   r   r   dump)r   r   r   r   r   r   r   r   r   r   r   $   s   
"r   c              
   C   sR   zt | }W |S  ty( } ztd|  t| }W Y d}~|S d}~ww )zWrapper for orjson.loads.zError using orjson.loads: N)r   loadsr   r   r   r   )r   decodedr   r   r   r   r   3   s   r   c              
   C   sV   z
t |  }W |S  ty* } ztd|  t| }W Y d}~|S d}~ww )zWrapper for orjson.load.zError using orjson.load: N)r   r   readr   r   r   r   load)r   r   r   r   r   r   r    =   s   r    )r   r   r    r   )r   loggingostypingr   r   	getLogger__name__r   r   environgetstrr   r   bytesr   r    ImportErrorr   r   r   r   <module>   s     

