
    NjZ                     4    d dl Z d dlmZ d Z G d de      Zy)    N)TorchDispatchModec                    t        | t        j                  t        j                  j                  f      rdt        | j                         dS t        | t
        t        f      rddj                  d | D              z   dz   S t        | t              r,ddj                  d | j                         D              z   d	z   S t        |       S )
NzT(s=)[z, c              3   2   K   | ]  }t        |        y w)Nto_str).0os     k/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/pocket_tts/utils/debugging.py	<genexpr>zto_str.<locals>.<genexpr>	   s     6#Qvay#s   ]{c              3   T   K   | ]   \  }}t        |       d t        |        " yw)z: Nr   )r
   kvs      r   r   zto_str.<locals>.<genexpr>   s(     S{tq!&)Bvayk:{s   &(})
isinstancetorchTensornn	Parameterlistshapetuplejoindictitemsstr)objs    r   r	   r	      s    #ehh&8&89:d399o&a((	C$	'TYY6#666<<	C	TYYSsyy{SSSVYYY3x    c                       e Zd ZdZddZy)LoggingModez+Useful to check implementation differences.Nc                      ||i |xs i }t        d| dt        |       dt        |       dt        |              |S )NzAten function called: z, args: z
, kwargs: z -> output: )printr	   )selffunctypesargskwargsoutputs         r   __torch_dispatch__zLoggingMode.__torch_dispatch__   sU    t,v|,$TF(d|nJvf~&6 7f~&(	

 r!   ) N)__name__
__module____qualname____doc__r,   r-   r!   r   r#   r#      s
    5r!   r#   )r   torch.utils._python_dispatchr   r	   r#   r-   r!   r   <module>r3      s     :
# 
r!   