o $i @s.ddlmZmZmZmZmZGdddZdS))AnyCallableListOptionalTuplec@seZdZdZdddedeedgdffddZdd Zd d Z e d e fd dZ e d efddZ d eeefddZdddZdS)BarrieraBarrier to collect results and process them in bulk. A barrier can be used to collect multiple results and process them in bulk once a certain count or a timeout is reached. For instance, if ``max_results=N``, the ``on_completion`` callback will be invoked once :meth:`arrive` has been called ``N`` times. The completion callback will only be invoked once, even if more results arrive after completion. The collected results can be resetted with :meth:`reset`, after which the callback may be invoked again. The completion callback should expect one argument, which is the barrier object that completed. Args: max_results: Maximum number of results to collect before a call to :meth:`wait` resolves or the :meth:`on_completion` callback is invoked. on_completion: Callback to invoke when ``max_results`` results arrived at the barrier. N) on_completion max_resultsrcCs||_d|_||_g|_dS)NF) _max_results _completed_on_completion_results)selfr rr`/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/air/execution/_internal/barrier.py__init__s zBarrier.__init__cGs,t|dkr |d}|j||dS)a,Notify barrier that a result successfully arrived. This will count against the ``max_results`` limit. The received result will be included in a call to :meth:`get_results`. Args: *data: Result data to be cached. Can be obtained via :meth:`get_results`. rN)lenr append_check_completion)rdatarrrarrive+s   zBarrier.arrivecCs8|jrdS|j|jkrd|_|jr||dSdSdS)NT)r num_resultsr r rrrrr;s zBarrier._check_completionreturncC|jS)z)Returns True if the barrier is completed.)r rrrr completedGszBarrier.completedcCs t|jS)z(Number of received (successful) results.)rr rrrrrLs zBarrier.num_resultscCr)z Return list of received results.)r rrrr get_resultsQszBarrier.get_resultscCsd|_g|_dS)a Reset barrier, removing all received results. Resetting the barrier will reset the completion status. When ``max_results`` is set and enough new events arrive after resetting, the :meth:`on_completion` callback will be invoked again. FN)r r rrrrresetUs z Barrier.reset)rN)__name__ __module__ __qualname____doc__intrrrrrpropertyboolrrrrrrrrrrrrs   rN)typingrrrrrrrrrrs