
    &`i	                     <   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	d dl
Z
e j                            dd          Ze j                            dd          Z	 dd	eeeeeef                           fd
Z	 dd	eeeeeef                           fdZed             ZdS )    N)contextmanager)ListOptionalTupleTESTS_TO_RUN TESTS_TO_SKIP   
   returnc           
      N   t           rt           | vrd gS t          rt          | v rd gS t          j        |           t          j                    }d}t          j                    |z
  dk     r) |             |dz  }t          j                    |z
  dk     )|dz  dz   }g }t          d          D ]}t          j                    }d}t          j                    |z
  dk     r;t          |          D ]} |             ||z  }t          j                    |z
  dk     ;t          j                    }	|                    ||z  |	|z
  z             t          j        |          }
t          j	        |          }t          | dt          |
d          dt          |d                     | |
|fgS Nr   r
   r         z
per secondz+-filter_patternskip_patterntimesleepperf_counterrangeappendnpmeanstdprintround)namefn
multiplierwarmup_time_secstartcountstepstats_endr   sds               {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/ray_microbenchmark_helpers.pytimeitr*      s     .44v ,,v 	JEE




%
)
)

 



%
)
) B;?DE1XX 9 9!##!!E)A--4[[  TME !!E)A-- !!Z%'3;7888875>>D	B	$eD!nndE"aLLAAA4    c           
      j  K   t           rt           | vrd gS t          rt          | v rd gS t          j        |           t          j                    }d}t          j                    |z
  dk     r/ |             d {V  |dz  }t          j                    |z
  dk     /|dz  dz   }g }t          d          D ]}t          j                    }d}t          j                    |z
  dk     rAt          |          D ]} |             d {V  ||z  }t          j                    |z
  dk     At          j                    }	|                    ||z  |	|z
  z             t          j        |          }
t          j	        |          }t          | dt          |
d          dt          |d                     | |
|fgS r   r   )r   async_fnr    r!   r"   r#   r$   r%   r&   r'   r   r(   s               r)   asyncio_timeitr.   3   s       .44v ,,v 	JEE




%
)
)hjj
 



%
)
) B;?DE1XX 9 9!##!!E)A--4[[ ! !hjj        TME !!E)A-- !!Z%'3;7888875>>D	B	$eD!nndE"aLLAAA4r+   c               +      K   t          j        di |  	 d V  t          j                     d S # t          j                     w xY w)N )rayinitshutdown)	init_argss    r)   ray_setup_and_teardownr5   V   sI      Hy


s	   . A)r
   r   )osr   
contextlibr   typingr   r   r   numpyr   r1   environgetr   r   strfloatr*   r.   r5   r0   r+   r)   <module>r>      s'   				  % % % % % % ( ( ( ( ( ( ( ( ( (     



 33z~~or22 -/   	(5eU*+
,-       H 35   	(5eU*+
,-       F     r+   