+
    ÒÜ³j1  ã                   óü  € ^ RI t ^ RIt^ RIt^ RIt^ RIt^ RIt^ RIt^ RIt^ RIt^ RI	t
^ RIHt ^ RIt^ RIHt ^ RIHtHt ^ RIHtHt ^ RIHtHtHtHtHt ^ RIHtHtHt R*R R	 llt]P@                  ! 4       R
 R l4       t!R+R R llt"]PF                  R R l4       t$]PF                  R 4       t%]PF                  R 4       t& ! R R]'4      t( ! R R] PR                  4      t* ! R R]PV                  4      t,]PF                  R 4       t-R R lt.R R lt/. R,O^R3R R llt0R+R  R! llt1]2R"8X  dJ   R#]Pf                  R$&   ^ R%I4H5t5 R&t6]5Pn                  ! ]6R'R(R)7      t8]Pr                  ! ]64      t:]! ]8]:4       R# R# )-é    N)Útqdm)Úload_dataset)ÚdefaultdictÚCounter)ÚThreadPoolExecutorÚas_completed)ÚListÚUnionÚIterableÚDictÚOptional)ÚAutoTokenizerÚPreTrainedModelÚPreTrainedTokenizerc                óH   € V ^8„  d   QhR\         R\        R\        R\        /# )é   ÚmodelÚ	tokenizerÚout_pathÚformat_tabs)r   r   ÚstrÚbool)Úformats   "Ú_/home/luca/screen-checking/.venv/lib/python3.14/site-packages/awq/evaluation/humaneval_utils.pyÚ__annotate__r   "   s0   € ÷ (ñ (Üð(ä"ð(ô ð(ô ñ	(ó    c                 ó¤  € \        R 4      R,           Uu/ uF  pVR,          VbK  	  pp. p\        \        VP                  4       4      \	        V4      R7       F_  w  pw  r‰V'       d   V	R,          P                  RR4      p
M	V	R,          p
\        WV
^4      pV F  p\        VVR7      pWm.,          pK  	  Ka  	  \        VR4      ;_uu_ 4       pV F>  pVP                  \        P                  ! V4      R	,           P                  R
4      4       K@  	  RRR4       \        V^.^RR7      p\        V4       R# u upi   + '       g   i     L2; i)Úopenai_humanevalÚtestÚtask_id©ÚtotalÚpromptú    Ú	)r    Ú
completionÚwbÚ
zutf-8Nç      @)Úsample_fileÚkÚ	n_workersÚtimeout)r   r   Ú	enumerateÚitemsÚlenÚreplaceÚgenerate_batch_completionÚdictÚopenÚwriteÚjsonÚdumpsÚencodeÚevaluate_functional_correctnessÚprint)r   r   r   r   ÚexampleÚproblemsÚsamplesÚir    Útaskr#   Úbatch_completionsÚsampleÚresultÚfpÚxÚresultss   &&&&             r   Úeval_humanevalrF   "   s7  € ô $Ð$6Ó7¸Ö?óá?ˆGð 	�	Õ˜GÒ#Ù?ð ð ð
 €Gä"¤9¨X¯^©^Ó-=Ó#>ÄcÈ(Ãm×TÑˆ‰?ˆGßØ˜(•^×+Ñ+¨F°DÓ9‰Fà˜(•^ˆFä5°eÈÐPQÓRÐã'ˆFÜØØ!ôˆFð
 �xÕŠGó (ñ Uô  
ˆh˜×	Ô	 ÛˆAØ�H‰H”d—j’j “m dÕ*×2Ñ2°7Ó;Ö<ñ ÷ 
ô .ØØˆ#ØØô	€Gô 
ˆ'†NùòE÷. 
×	ús   •D:ÃAD?Ä?E	c                óR   € V ^8„  d   QhR\         R\        R\        \        ,          /# )r   r   r   Úreturn)r   r   Úlistr   )r   s   "r   r   r   N   s-   € ÷ Vñ VÜðVÜ':ðVä	Œ#…YñVr   c                 óì  € \        V4       Uu. uF  qBNK  	  ppV! VR R7      P                  V P                  P                  4      pVP                  P                  ^R7      pV P                  ! R/ VBRRRRRRR	R
RRRVP                  RVP                  /B pTP                  V U	u. uF  q™VR NK	  	  up	RR7      p
R R lpR R lpV
 Uu. uF  qÛ! V! V4      4      NK  	  up# u upi u up	i u upi )Úpt)Úreturn_tensors)ÚdimÚ	use_cacheTÚmax_seq_leni   Útemperaturegš™™™™™É?Útop_pgffffffî?Ú	do_sampleÚeos_token_idÚpad_token_idN)Úskip_special_tokensc                ó0   € V ^8„  d   QhR\         R\         /# )r   r&   rH   ©r   )r   s   "r   r   Ú/generate_batch_completion.<locals>.__annotate__e   s   € ÷ +ñ +¤ð +¬ñ +r   c                 óT   € V P                  R 4      p V P                  R4      ^ ,          # )r(   z

)ÚlstripÚsplit)r&   s   &r   Úfilter_codeÚ.generate_batch_completion.<locals>.filter_codee   s(   € à×&Ñ& tÓ,ˆ
Ø×Ñ Ó'¨Õ*Ð*r   c                ó0   € V ^8„  d   QhR\         R\         /# )r   ÚtextrH   rW   )r   s   "r   r   rX   j   s   € ÷ *ñ *œ#ð *¤#ñ *r   c                 ó&   € V P                  R R4      # )r%   r$   )r1   )r_   s   &r   Úfix_indentsÚ.generate_batch_completion.<locals>.fix_indentsj   s   € Ø�|‰|˜D &Ó)Ð)r   © )	ÚrangeÚtor   ÚdeviceÚ	input_idsÚsizeÚgeneraterS   Úbatch_decode)r   r   r#   Ú
batch_sizeÚ_Úinput_batchÚinputsÚinput_ids_cutoffÚgenerated_idsÚidsr@   r\   ra   r&   s   &&&&          r   r2   r2   M   s3  € ô $)¨Ô#4Ó5Ñ#4˜a’6Ñ#4€KÐ5Ù�{°4Ô8×;Ñ;¸E¿K¹K×<NÑ<NÓO€FØ×'Ñ'×,Ñ,°Ð,Ó3Ðà—N’Nñ 	Ø
ñ	àð	ð ð	ð ð		ð
 ð	ð ð	ð ×+Ñ+ð	ð ×+Ñ+ò	€Mð "×.Ñ.Ù+8Ó9©= CÐÐÓ	©=Ñ9Ø ð /ó Ðõ
+õ
*ñ DUÓUÑCT°ZˆK™ JÓ/Ö0ÑCTÑUÐUùò9 6ùò  	:ùò Vs   Ž
C'Â(C,ÃC1c          
      ój   € V ^8„  d   QhR\         R\        R\        R\        \        ,          R\         /# )r   Úproblemr&   r-   Úcompletion_idrH   )r   r   Úfloatr   Úint)r   s   "r   r   r   p   s?   € ÷ Cñ CÜðCÜ"ðCÜ-2ðCÜCKÌCÅ=ðCä	ñCr   c                ó²  a aaa€ VV VV3R lp\         P                  ! 4       pVP                  4       o\         P                  ! VR7      pVP	                  4        VP                  S^,           R7       VP                  4       '       d   VP                  4        S'       g   SP                  R4       \        S R,          S^ ,          R8H  S^ ,          VR7      # )zë
Evaluates the functional correctness of a completion by running the test
suite provided in the problem.

:param completion_id: an optional completion ID so we can match
    the results later even if execution finishes asynchronously.
c                  óü  <€ \        4       ;_uu_ 4        ^ RIp ^ RIpVP                  pV P                  pV P
                  p\        4        S	R,          S,           R,           S	R,          ,           R,           RS	R,           R2,           p / p\        4       ;_uu_ 4        \        S4      ;_uu_ 4        \        WV4       RRR4       RRR4       S
P                  R4       W!n        W0n        W@n        RRR4       R#   + '       g   i     LE; i  + '       g   i     LP; i  \         d    S
P                  R	4        L`\         d   pS
P                  R
T 24        Rp?LƒRp?ii ; i  + '       g   i     R# ; i)r   Nr#   r(   r   zcheck(Úentry_pointÚ)Úpassedú	timed outzfailed: )Úcreate_tempdirÚosÚshutilÚrmtreeÚrmdirÚchdirÚreliability_guardÚ
swallow_ioÚ
time_limitÚexecÚappendÚTimeoutExceptionÚBaseException)r~   r   r€   r�   r‚   Úcheck_programÚexec_globalsÚer&   rs   rB   r-   s           €€€€r   Úunsafe_executeÚ)check_correctness.<locals>.unsafe_execute{   s2  ø€ Ü×ÕãÛà—]‘]ˆFØ—H‘HˆEØ—H‘HˆEô Ôð ˜Õ!Øõàõð ˜&•/õ"ð õ	ð
 ˜7 =Õ1Ð2°!Ð4õ5ð ð	.Ø!�Ü—\•\Ü# G×,Õ,Ü˜]Ô9÷ -÷ "ð —‘˜hÔ'ð #ŒMØŒHØŒH÷I Ñ÷2 -×,ú÷ "—\ûô $ô +Ø—‘˜kÖ*Ü ô .Ø—‘ ¨¨˜n×-Ñ-ûð.ú÷= ×Ðúsx   ”A5E*Â
D!ÂD	Â2C;Â>D	ÃD!ÃE*Ã;DÄD	ÄDÄD!Ä!E'Ä=E*Ä?E'ÅE'Å	E"ÅE*Å"E'Å'E*Å*E;	)Útarget)r-   r|   r    r{   )r    r{   rB   rt   )
ÚmultiprocessingÚManagerrI   ÚProcessÚstartÚjoinÚis_aliveÚkillr‡   r3   )rs   r&   r-   rt   r�   ÚmanagerÚprB   s   fff&   @r   Úcheck_correctnessr™   p   s¢   û€ ÷%ð %ôN ×%Ò%Ó'€GØ�\‰\‹^€Fä×Ò ~Ô6€AØ‡G�G„IØ‡F�F�7˜Q•;€FÔØ‡z�z‡|‚|Ø	�‰ŒçØ�‰�kÔ"äØ˜	Õ"Ø�a�y˜HÑ$Ø�a�yØ#ô	ð r   c                ó$   € V ^8„  d   QhR\         /# )r   Úseconds)ru   )r   s   "r   r   r   ·   s   € ÷ 	0ñ 	0œñ 	0r   c              #   óP  "  € R  p\         P                  ! \         P                  V 4       \         P                   ! \         P                  V4        Rx € \         P                  ! \         P                  ^ 4       R#   \         P                  ! \         P                  ^ 4       i ; i5i)c                 ó   € \        R 4      h)z
Timed out!)rˆ   )ÚsignumÚframes   &&r   Úsignal_handlerÚ"time_limit.<locals>.signal_handler¸   s   € Ü˜|Ó,Ð,r   N)ÚsignalÚ	setitimerÚITIMER_REALÚSIGALRM)r›   r    s   & r   r…   r…   ¶   sh   é € ò-ô ×Ò”V×'Ñ'¨Ô1Ü
‡M‚M”&—.‘. .Ô1ð0Ûä×Òœ×+Ñ+¨QÖ/øŒ×Òœ×+Ñ+¨QÕ/üs   ‚AB&ÁA< Á'B&Á<'B#Â#B&c               #   ór  "  € \        4       p \        P                  ! V 4      ;_uu_ 4        \        P                  ! V 4      ;_uu_ 4        \	        V 4      ;_uu_ 4        R x € R R R 4       R R R 4       R R R 4       R #   + '       g   i     L"; i  + '       g   i     L-; i  + '       g   i     R # ; i5i©N)ÚWriteOnlyStringIOÚ
contextlibÚredirect_stdoutÚredirect_stderrÚredirect_stdin)Ústreams    r   r„   r„   Ã   sj   é € äÓ €FÜ	×	#Ò	# F×	+Õ	+Ü×'Ò'¨×/Õ/Ü ×'Õ'Û÷ (÷ 0÷ 
,Ñ	+ç'×'ú÷ 0×/ú÷ 
,×	+Ð	+üsL   ‚)B7«B#Á
B	ÁA=Á#B	Á+B#Á3
B7Á=BÂB	ÂB ÂB#Â#B4	Â.	B7c            	   #   óè   "  € \         P                  ! 4       ;_uu_ 4       p \        V 4      ;_uu_ 4        V x € R R R 4       R R R 4       R #   + '       g   i     L; i  + '       g   i     R # ; i5ir§   )ÚtempfileÚTemporaryDirectoryr‚   )Údirnames    r   r}   r}   Ì   sB   é € ä	×	$Ò	$×	&Ô	&¨'Ü�7�^�^ØŠM÷ ÷ 
'Ñ	&ß�^ú÷ 
'×	&Ð	&üs2   ‚A2 A´A	¹AÁ
A2ÁAÁAÁA/	Á)	A2c                   ó   € ] tR t^ÓtRtR# )rˆ   rc   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__static_attributes__rc   r   r   rˆ   rˆ   Ó   s   † Ûr   rˆ   c                   ó<   a € ] tR t^×t o RtR tR tR tR tRt	V t
R# )r¨   z5StringIO that throws an exception when it's read fromc                ó   € \         hr§   ©ÚIOError©ÚselfÚargsÚkwargss   &*,r   ÚreadÚWriteOnlyStringIO.readÚ   ó   € Üˆr   c                ó   € \         hr§   rº   r¼   s   &*,r   ÚreadlineÚWriteOnlyStringIO.readlineÝ   rÂ   r   c                ó   € \         hr§   rº   r¼   s   &*,r   Ú	readlinesÚWriteOnlyStringIO.readlinesà   rÂ   r   c                ó   € R# )z*Returns True if the IO object can be read.Frc   r¼   s   &*,r   ÚreadableÚWriteOnlyStringIO.readableã   s   € ár   rc   N)r³   r´   rµ   r¶   Ú__doc__rÀ   rÄ   rÇ   rÊ   r·   Ú__classdictcell__)Ú__classdict__s   @r   r¨   r¨   ×   s!   ø‡ € Ù?òòò÷ð r   r¨   c                   ó   € ] tR t^ètRtRtR# )r¬   Ústdinrc   N)r³   r´   rµ   r¶   Ú_streamr·   rc   r   r   r¬   r¬   è   s   † Ø„Gr   r¬   c              #   ó  "  € V R 8X  d   Rx € R# \         P                  ! 4       p\         P                  ! V 4        Rx €  \         P                  ! T4       R#   \         d   pThRp?ii ; i  \         P                  ! T4       i ; i5i)Ú.N)r~   Úgetcwdr‚   r‰   )ÚrootÚcwdÚexcs   &  r   r‚   r‚   ì   sa   é € àˆs„{ÛÙÜ
�)Š)‹+€CÜ‡H‚HˆT„NðÜô 	�Š�Žøô ô Øˆ	ûðûô 	�Š��üs4   ‚9B	¼A ÁB	ÁA+Á$A&Á&A+Á+A. Á.BÂB	c                óF   € V ^8„  d   QhR\         R\        \        ,          /# )r   ÚfilenamerH   )r   r   r   )r   s   "r   r   r   û   s   € ÷ +ñ +œ3ð +¤8¬D¥>ñ +r   c           	   #  óî  "  € V P                  R4      '       d¥   \        V R4      ;_uu_ 4       p\        P                  ! VR4      ;_uu_ 4       pV FW  p\        ;QJ d    R V 4       F  '       g   K   RM	  RM! R V 4       4      '       g   K?  \        P
                  ! V4      x € KY  	  RRR4       RRR4       R# \        V R4      ;_uu_ 4       pV FW  p\        ;QJ d    R	 V 4       F  '       g   K   RM	  RM! R	 V 4       4      '       g   K?  \        P
                  ! V4      x € KY  	  RRR4       R#   + '       g   i     L–; i  + '       g   i     R# ; i  + '       g   i     R# ; i5i)
z6
Parses each jsonl line and yields it as a dictionary
z.gzÚrbÚrtc              3   óJ   "  € T F  qP                  4       '       * x € K  	  R # 5ir§   ©Úisspace©Ú.0rD   s   & r   Ú	<genexpr>Ústream_jsonl.<locals>.<genexpr>  s   é € Ð9±D¨qŸy™y›{Ÿ?™?³Dùó   ‚!#TFNÚrc              3   óJ   "  € T F  qP                  4       '       * x € K  	  R # 5ir§   rÞ   rà   s   & r   râ   rã     s   é € Ð5±¨1Ÿ9™9›;—‘³ùrä   )Úendswithr4   ÚgzipÚanyr6   Úloads)rÙ   ÚgzfprC   Úlines   &   r   Ústream_jsonlrí   û   sô   é € ð ×Ñ˜×ÒÜ�(˜D×!Ô! TÜ—’˜4 ×&Ô&¨"Û�Dß“sÑ9±DÓ9—s—s’sÑ9±DÓ9×9Ô9Ü"Ÿjšj¨Ó.Ô.ñ ÷ '÷ "Ñ!ô �(˜C× Ô  BÛ�ß“3Ñ5±Ó5—3—3’3Ñ5±Ó5×5Ô5ÜŸ*š* TÓ*Ô*ñ ÷ !Ñ ÷ '×&ú÷ "×!Ð!ú÷ !× Ð üsd   ‚,E5® EÁ"D:	Á3D:	ÂD:	Â,EÂ4E5Ã"E!Ã7E!ÄE!Ä0
E5Ä:E
ÅEÅE	Å	E5Å!E2	Å,	E5c                óþ   € V ^8„  d   QhR\         \        \        \        ,          \        P                  3,          R\         \        \        ,          \        P                  3,          R\        R\        P                  /# )r   Únum_samplesÚnum_correctr+   rH   )r
   rv   r	   ÚnpÚndarray)r   s   "r   r   r     sZ   € ÷ ñ Ü”sœD¤�I¤r§z¡zÐ1Õ2ðä”tœC•y¤"§*¡*Ð,Õ-ðô ðô ‡Z�Zñ	r   c                ón  € R R lp\        V \        4      '       d"   \        P                  ! V \	        V4      4      pM&\	        V 4      \	        V4      8X  g   Q h\        V 4      p\        P                  ! \        WA4       UUu. uF!  w  rVV! \        V4      \        V4      V4      NK#  	  upp4      # u uppi )z@
Estimates pass@k of each problem and returns them in an array.
c                óH   € V ^8„  d   QhR\         R\         R\         R\        /# )r   ÚnÚcr+   rH   )rv   ru   )r   s   "r   r   Ú(estimate_pass_at_k.<locals>.__annotate__  s.   € ÷ Dñ D”Sð DœSð D¤Sð D¬Uñ Dr   c           
     óÈ   € W,
          V8  d   R# R\         P                  ! RV\         P                  ! W,
          ^,           V ^,           4      ,          ,
          4      ,
          # )z-
Calculates 1 - comb(n - c, k) / comb(n, k).
g      ð?)rñ   ÚprodÚarange)rõ   rö   r+   s   &&&r   Ú	estimatorÚ%estimate_pass_at_k.<locals>.estimator  sC   € ð �5�1Œ9ÙØ”R—W’W˜S 1¤r§y¢y°µ¸µ¸AÀ½EÓ'BÕ#BÕBÓCÕCÐCr   )	Ú
isinstancerv   Ú	itertoolsÚrepeatr0   Úiterrñ   ÚarrayÚzip)rï   rð   r+   rû   Únum_samples_itrõ   rö   s   &&&    r   Úestimate_pass_at_kr    s�   € õDô �+œs×#Ò#Ü"×)Ò)¨+´s¸;Ó7GÓH‰ä�;Ó¤3 {Ó#3Ô3Ð3Ð3Ü˜kÓ*ˆä�8Š8Ü25°nÔ2RÔSÑ2R©$¨!‰”3�q“6œ3˜q›6 1Ö	%Ñ2RÒSóð ùÛSs   Â'B1
r)   c                ó^   € V ^8„  d   QhR\         R\        \        ,          R\        R\        /# )r   r*   r+   r,   r-   )r   r	   rv   ru   )r   s   "r   r   r   (  s:   € ÷ Dñ DÜðDäŒC…yðDô ðDô ñ	Dr   c           
      óh  a a€ \        R 4      R,           Uu/ uF  pVR,          VbK  	  pp\        VR7      ;_uu_ 4       p. p\        4       p^ p	\        \        4      o\        R4       \        \        S 4      4       Fl  p
V
R,          pV
R,          pW[,          WÃW‹,          3pVP                  ! \        .VO5!  pVP                  V4       W‹;;,          ^,          uu&   V	^,          p	Kn  	  \        V4      \        V4      8  dw   \	        VP                  4       4      R\        V4       p\        R\        V4       R\        V4       R	24       VP                  4        UUu/ uF  w  ppW9   g   K  VVbK  	  ppp\        V4      \        V4      8X  g   Q R
4       h\        R4       \        \        V4      \        V4      R7       F;  pVP                  4       pSVR,          ,          P                  VR,          V34       K=  	  RRR4       . . ppSP!                  4        Ff  pVP#                  4        V Uu. uF  pV^,          R,          NK  	  ppVP                  \        V4      4       VP                  \%        V4      4       Kh  	  \&        P(                  ! V4      p\&        P(                  ! V4      pXpV Uu/ uF=  pVV8¬  P+                  4       '       g   K  RV 2\-        VVV4      P/                  4       bK?  	  ppVV 3R lpV# u upi u uppi   + '       g   i     EL; iu upi u upi )r   r   r    )Úmax_workerszReading samples...r&   NzOnly found z# solutions, reducing problems from z...z Some problems are not attempted.zRunning test suites...r!   rt   r{   zpass@c               3   óÊ   <"  € \        S4       FN  p V R ,          pSV,          P                  ^ 4      pV^,          R,          V R&   V^,          R,          V R&   V x € KP  	  R# 5i)r    rB   r{   N)rí   Úpop)rA   r    rB   rE   r*   s      €€r   Úcombine_resultsÚ8evaluate_functional_correctness.<locals>.combine_resultsd  s_   øé € Ü" ;Ö/ˆFØ˜YÕ'ˆGØ˜WÕ%×)Ñ)¨!Ó,ˆFØ% a�y¨Õ2ˆF�8ÑØ% a�y¨Õ2ˆF�8ÑØŒLó 0ùs   ƒA A#)r   r   r   r   rI   r:   r   rí   Úsubmitr™   r‡   r0   Úkeysr/   r   rB   ÚvaluesÚsortÚsumrñ   r  Úallr  Úmean)r*   r+   r,   r-   r;   r<   ÚexecutorÚfuturesrt   Ú	n_samplesrA   r    r&   r¾   ÚfutureÚinclude_keysÚvrB   r"   Úcorrectrå   r{   ÚksÚ	pass_at_kr
  rE   s   f&&&                     @r   r9   r9   (  sà  ù€ ô $Ð$6Ó7¸Ö?óá?ˆGð 	�	Õ˜GÒ#Ù?ð ð ô 
¨	×	2Õ	2°hØˆÜ›	ˆØˆ	ÜœdÓ#ˆäÐ"Ô#Üœ<¨Ó4Ö5ˆFØ˜YÕ'ˆGØ Õ-ˆJØÕ% z¸MÕ<RÐSˆDØ—_’_Ô%6Ð>¸Ó>ˆFØ�N‰N˜6Ô"Ø×" aÕ'Ó"Ø˜�NŠIñ 6ô ˆ}Ó¤ H£Ô-Ü §¡£Ó0Ð1E´3°}Ó3EÐFˆLÜØœc -Ó0Ð1Ð1TÔUXÐYaÓUbÐTcÐcfÐgôð *2¯©Ô)9ÔOÑ)9¡  A¸QÑ=Nœ˜˜1šÑ)9ˆHÑOä�=Ó!¤S¨£]Ô2ÐVÐ4VÓVÐ2äÐ&Ô'Üœ<¨Ó0¼¸G»×EˆFØ—]‘]“_ˆFØ�F˜9Õ%Õ&×-Ñ-¨v°oÕ/FÈÐ.OÖPñ F÷5 
3ð> ˜ˆ7€EØ—.‘.Ö"ˆØ�‰ŒÙ*0Ó1©& Q�!�A•$�x—.�.©&ˆÐ1Ø�‰”S˜“[Ô!Ø�‰”s˜6“{Ö#ñ	 #ô
 �HŠH�U‹O€EÜ�hŠh�wÓ€Gà	
€Bñ óáˆAØ�Q‰J×Ñ×ô 	Bˆ%�ˆsˆÔ'¨¨w¸Ó:×?Ñ?ÓAÒAÙð ð öð Ðùò}ùó6 P÷+ 
3×	2Ð	2üòD 2ùòs=   —LÁ D"LÅ"L
Å2L
Å9BLÈ8L*Ê=L/Ë#L/ÌLÌL'	c                ó:   € V ^8„  d   QhR\         \        ,          /# )r   Úmaximum_memory_bytes)r   rv   )r   s   "r   r   r   o  s   € ÷ T"ñ T"¬H´S­Mñ T"r   c                óÜ  € ^ RI p^ RIpV e{   ^ RIpVP                  VP                  W 34       VP                  VP
                  W 34       VP                  4       P                  R8X  g   VP                  VP                  W 34       VP                  4        ^ RI
pRVn        RVn        ^ RIpRVP                  R&   RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn        RVn         RVn!        RVn"        RVn        RVn#        RVn$        RVn%        RVn&        RVn'        ^ RI(pRVn)        RVn*        RVn!        ^ RI+pRVn,        ^ RI-pRVP\                  R&   RVP\                  R&   RVP\                  R&   RVP\                  R&   RVP\                  R	&   R# )
aª  
This disables various destructive functions and prevents the generated code
from interfering with the test (e.g. fork bomb, killing other processes,
removing filesystem files, etc.)

WARNING
This function is NOT a security sandbox. Untrusted code, including, model-
generated code, should not be blindly executed outside of one. See the
Codex paper for more information about OpenAI's code sandbox, and proceed
with caution.
NÚDarwinÚ1ÚOMP_NUM_THREADSÚipdbÚjoblibÚresourceÚpsutilÚtkinter)/ÚplatformÚfaulthandlerr$  Ú	setrlimitÚ	RLIMIT_ASÚRLIMIT_DATAÚunameÚsystemÚRLIMIT_STACKÚdisableÚbuiltinsÚexitÚquitr~   Úenvironr–   ÚputenvÚremoveÚ
removedirsr�   ÚfchdirÚsetuidÚforkÚforkptyÚkillpgÚrenameÚrenamesÚtruncater1   ÚunlinkÚfchmodÚfchownÚchmodÚchownÚchrootÚlchflagsÚlchmodÚlchownrÔ   r‚   r   r€   ÚmoveÚ
subprocessÚPopenÚsysÚmodules)	r  r'  r(  r$  r0  r~   r   rI  rK  s	   &        r   rƒ   rƒ   o  sÚ  € ÷ "àÒ'Ûà×ÑØ×ÑÐ!5Ð Lô	
ð 	×ÑØ× Ñ Ð#7Ð"Nô	
ð �~‰~Ó×&Ñ&¨(Ô2Ø×ÑØ×%Ñ%Ð(<Ð'Sôð ×ÑÔãà€H„MØ€H„Mãà$'€B‡J�JÐ Ñ!à€B„GØ€B„IØ€B„IØ€B„IØ€B„MØ€B„HØ€B„IØ€B„IØ€B„GØ€B„JØ€B„IØ€B„IØ€B„JØ€B„KØ€B„JØ€B„IØ€B„IØ€B„IØ€B„HØ€B„HØ€B„IØ€B„IØ€B„KØ€B„IØ€B„IØ€B„IØ€B„Hãà€F„MØ€F„KØ€F„Lãà€JÔãà€C‡K�K�ÑØ €C‡K�K�ÑØ"€C‡K�K�
ÑØ €C‡K�K�ÑØ!€C‡K�K�	Ór   Ú__main__ÚtrueÚTOKENIZERS_PARALLELISM)ÚAutoAWQForCausalLMzTheBloke/zephyr-7B-beta-AWQÚautoi   )Ú
device_maprO   )zhumaneval_out.jsonlTr§   )é   é
   éd   );Úior~   r6   rè   Útorchr¢   r¯   rþ   r©   Únumpyrñ   r   r�   Údatasetsr   Úcollectionsr   r   Úconcurrent.futuresr   r   Útypingr	   r
   r   r   r   Útransformersr   r   r   rF   Úinference_moder2   r™   Úcontextmanagerr…   r„   r}   Ú	Exceptionrˆ   ÚStringIOr¨   Ú_RedirectStreamr¬   r‚   rí   r  r9   rƒ   r³   r3  ÚawqrP  Ú
model_pathÚfrom_quantizedr   Úfrom_pretrainedr   rc   r   r   Ú<module>rg     s‡  ðó 
Û 	Û Û Û Û Û Û Û Û Ý Û Ý !ß ,ß ?÷õ ÷ñ ÷(ðV ×ÒÓôVó ðV÷DCðL ×Ñô	0ó ð	0ð ×Ññó ðð ×Ññó ðô	�yô 	ô˜Ÿ™ô ô"�Z×/Ñ/ô ð ×Ññó ðõ+õ"ò<  ØØ÷	D÷NT"ðn ˆzÔØ+1€B‡J�JÐ'Ñ(å&à.€JØ×-Ò-Ø˜v°4ô€Eð ×-Ò-¨jÓ9€IÙ�5˜)Ö$ñ r   