+
    TV-js  ã                   óº   € ^ RI t  ^ RIt ^ RIt ! R R4      t ! R R]4      t ! R R]4      tR]R	]/tRR
 R lltR#   ] d    Rt L@i ; i  ] d    Rt LJi ; i)é    Nc                   óD   a € ] tR t^t o V 3R lR ltV 3R lR ltRtV tR# )ÚTrainingCallbackc                ó    <€ V ^8„  d   QhRS[ /# ©é   Ú
train_info©Údict)ÚformatÚ__classdict__s   "€Úg/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_lm/tuner/callbacks.pyÚ__annotate__ÚTrainingCallback.__annotate__   s   ø€ ÷ ñ ©tñ ó    c                ó   € R# )z6Called to report training loss at specified intervals.N© ©Úselfr   s   &&r   Úon_train_loss_reportÚ%TrainingCallback.on_train_loss_report   ó   € ár   c                ó    <€ V ^8„  d   QhRS[ /# ©r   Úval_infor	   )r   r   s   "€r   r   r      s   ø€ ÷ ñ ©4ñ r   c                ó   € R# )zICalled to report validation loss at specified intervals or the beginning.Nr   ©r   r   s   &&r   Úon_val_loss_reportÚ#TrainingCallback.on_val_loss_report   r   r   r   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r   Ú__static_attributes__Ú__classdictcell__©r   s   @r   r   r      s   ø‡ € ÷ð ÷ö r   r   c                   ól   a € ] tR t^t o RV 3R lR lltV 3R lR ltV 3R lR ltV 3R lR	 ltR
tV t	R# )ÚWandBCallbackNc                ó2   <€ V ^8„  d   QhRS[ RS[ RS[RS[/# ©r   Úproject_nameÚlog_dirÚconfigÚwrapped_callback©Ústrr
   r   )r   r   s   "€r   r   ÚWandBCallback.__annotate__   ó3   ø€ ÷ 
ñ 
áð
ñ ð
ñ ð	
ñ
 +ñ
r   c                ó¦   € \         f   \        R4      hW@n        \         P                  ! V\        P
                  P                  V4      VVR7       R # )NzCwandb is not installed. please install wandb via: pip install wandb)ÚprojectÚnameÚdirr,   )ÚwandbÚImportErrorr-   ÚinitÚosÚpathÚbasename©r   r*   r+   r,   r-   s   &&&&&r   Ú__init__ÚWandBCallback.__init__   sH   € ô Š=ÜØUóð ð !1ÔÜ�
Š
Ø Ü—‘×!Ñ! 'Ó*ØØ÷		
r   c                ó&   <€ V ^8„  d   QhRS[ RS[ /# ©r   ÚdataÚreturnr	   )r   r   s   "€r   r   r0   /   ó   ø€ ÷ Wñ W©Tð W±dñ Wr   c           	     óœ   € VP                  4        UUu/ uF*  w  r#T\        VR 4      '       d   VP                  4       MTbK,  	  upp# u uppi ©Útolist©ÚitemsÚhasattrrF   ©r   rA   ÚkÚvs   &&  r   Ú_convert_to_serializableÚ&WandBCallback._convert_to_serializable/   ó;   € ØIMÏÉÌÔVÉÁÀ�¤¨¨H×!5Ò!5�1—8‘8”:¸1Ò<ÉÒVÐVùÓVó   ”0Ac                ó    <€ V ^8„  d   QhRS[ /# r   r	   )r   r   s   "€r   r   r0   2   ó   ø€ ÷ Cñ C©tñ Cr   c                óÐ   € \         P                  ! V P                  V4      VP                  R 4      R7       V P                  '       d   V P                  P                  V4       R# R# ©Ú	iteration)ÚstepN)r6   ÚlogrM   Úgetr-   r   r   s   &&r   r   Ú"WandBCallback.on_train_loss_report2   sN   € Ü�	Š	Ø×)Ñ)¨*Ó5¸J¿N¹NÈ;Ó<Wõ	
ð × × Ð Ø×!Ñ!×6Ñ6°zÖBñ !r   c                ó    <€ V ^8„  d   QhRS[ /# r   r	   )r   r   s   "€r   r   r0   9   ó   ø€ ÷ ?ñ ?©4ñ ?r   c                óÐ   € \         P                  ! V P                  V4      VP                  R 4      R7       V P                  '       d   V P                  P                  V4       R# R# rT   )r6   rW   rM   rX   r-   r   r   s   &&r   r   Ú WandBCallback.on_val_loss_report9   sN   € Ü�	Š	Ø×)Ñ)¨(Ó3¸(¿,¹,À{Ó:Sõ	
ð × × Ð Ø×!Ñ!×4Ñ4°XÖ>ñ !r   ©r-   ©N©
r   r    r!   r"   r=   rM   r   r   r#   r$   r%   s   @r   r'   r'      ó1   ø‡ € ÷
ò 
÷&Wð W÷Cð C÷?ö ?r   r'   c                   ól   a € ] tR t^At o RV 3R lR lltV 3R lR ltV 3R lR ltV 3R lR	 ltR
tV t	R# )ÚSwanLabCallbackNc                ó2   <€ V ^8„  d   QhRS[ RS[ RS[RS[/# r)   r.   )r   r   s   "€r   r   ÚSwanLabCallback.__annotate__B   r1   r   c                óâ   € \         f   \        R4      hW@n        \         P                  ! V\        P
                  P                  V4      \        P
                  P                  VR4      VR7       R # )NzIswanlab is not installed. please install swanlab via: pip install swanlabÚswanlog)r3   Úexperiment_nameÚlogdirr,   )Úswanlabr7   r-   r8   r9   r:   r;   Újoinr<   s   &&&&&r   r=   ÚSwanLabCallback.__init__B   sW   € ô Š?ÜØ[óð ð !1ÔÜ�ŠØ ÜŸG™G×,Ñ,¨WÓ5Ü—7‘7—<‘< ¨Ó3Ø÷		
r   c                ó&   <€ V ^8„  d   QhRS[ RS[ /# r@   r	   )r   r   s   "€r   r   re   U   rC   r   c           	     óœ   € VP                  4        UUu/ uF*  w  r#T\        VR 4      '       d   VP                  4       MTbK,  	  upp# u uppi rE   rG   rJ   s   &&  r   rM   Ú(SwanLabCallback._convert_to_serializableU   rO   rP   c                ó    <€ V ^8„  d   QhRS[ /# r   r	   )r   r   s   "€r   r   re   X   rR   r   c                óÐ   € \         P                  ! V P                  V4      VP                  R 4      R7       V P                  '       d   V P                  P                  V4       R# R# rT   )rj   rW   rM   rX   r-   r   r   s   &&r   r   Ú$SwanLabCallback.on_train_loss_reportX   sN   € Ü�ŠØ×)Ñ)¨*Ó5¸J¿N¹NÈ;Ó<Wõ	
ð × × Ð Ø×!Ñ!×6Ñ6°zÖBñ !r   c                ó    <€ V ^8„  d   QhRS[ /# r   r	   )r   r   s   "€r   r   re   _   r[   r   c                óÐ   € \         P                  ! V P                  V4      VP                  R 4      R7       V P                  '       d   V P                  P                  V4       R# R# rT   )rj   rW   rM   rX   r-   r   r   s   &&r   r   Ú"SwanLabCallback.on_val_loss_report_   sN   € Ü�ŠØ×)Ñ)¨(Ó3¸(¿,¹,À{Ó:Sõ	
ð × × Ð Ø×!Ñ!×4Ñ4°XÖ>ñ !r   r^   r_   r`   r%   s   @r   rc   rc   A   ra   r   rc   r6   rj   c                óH   € V ^8„  d   QhR\         R\         R\         R\         /# )r   Ú	report_tor*   r+   r,   )r/   )r   s   "r   r   r   m   s0   € ÷ ñ Üðäðô ðô ñ	r   c           
      ó   € V e   V R8X  d   R # V P                  R4       Uu. uF8  qDP                  4       '       g   K  VP                  4       P                  4       NK:  	  p pR pV  F  p \        V,          ! VVVVR7      pK  	  V# u upi   \         d8   p\        T RRP                  \        P                  4       4       24      ThR p?ii ; i)NÚ Ú,)r*   r+   r,   r-   z$ callback doesn't exist choose from z, )ÚsplitÚstripÚlowerÚSUPPORT_CALLBACKÚKeyErrorÚ
ValueErrorrk   Úkeys)rw   r*   r+   r,   ÚitemÚtraining_callbackÚcallbackÚes   &&&&    r   Úget_reporting_callbacksr†   m   sÖ   € ð Ò˜I¨œOÙØ2;·/±/À#Ô2FÓWÑ2F¨$Ï*É*Ï,Ô%�—‘“×#Ñ#Ö%Ñ2F€IÐWØÐÛˆð	Ü 0°Ö :Ø)ØØØ!2ô	!Òñ ð Ðùò! Xøô ô 	ÜØ�*ð Ø#Ÿy™yÔ)9×)>Ñ)>Ó)@ÓAÐBðDóð ðûð	ús"   ¡B½"BÁ)BÂCÂ2CÃC)NNNN)	r9   r6   r7   rj   r   r'   rc   r~   r†   r   r   r   Ú<module>r‡      sŠ   ðó 
ðÛðÛ÷
ñ ô#?Ð$ô #?ôL#?Ð&ô #?ðN ˆ]ØˆðÐ ÷ñ øðM ô Ø‚Eðûð
 ô Ø‚Gðús    †? ‹A ¿	AÁ
AÁ	AÁA