i`hSSKJr SSKJr SSKrSSKJr SSKJr SSK J r SSK J r "SS 5r g) ) annotations)SequenceN)Producer)report_collection_diff)parse_tx_spec_config)WorkerControllerc\rSrSrSrSSSjjr\SSj5r\SSj5r\SSj5r SSjr SS jr SSS jjr SS jr SS jrSS jrSSjrSrg)EachScheduling aImplement scheduling of test items on all nodes. If a node gets added after the test run is started then it is assumed to replace a node which got removed before it finished its collection. In this case it will only be used if a node with the same spec got removed earlier. Any nodes added after the run is started will only get items assigned if a node with a matching spec was removed before it finished all its pending items. The new node will then be assigned the remaining items from the removed node. NcXl[[U55Ul0Ul0Ul/Ul0UlUc[S5Ul OURUl SUl g)N eachschedF) configlenrnumnodesnode2collection node2pending_started_removed2pendingrlogr collection_is_completed)selfrrs S/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/xdist/scheduler/each.py__init__EachScheduling.__init__s\ 089 BD?A02 CE ; ,DH}}DH',$cH[URR55$)z%A list of all nodes in the scheduler.)listrkeys)rs rnodesEachScheduling.nodes(sD%%**,--rcUR(dgUR(agURR5Hn[ U5S:dM g g)NFT)rrrvaluesrrpendings rtests_finishedEachScheduling.tests_finished-sG++  ((//1G7|q 2rcXURR5H nU(dM g g)zReturn True if there are pending test items. This indicates that collection has finished and nodes are still processing test items, so this can be thought of as "the scheduler is active". TF)rr#r$s r has_pendingEachScheduling.has_pending8s*((//1Gw2rcDXR;de/URU'gN)r)rnodes radd_nodeEachScheduling.add_nodeEs$,,,,,"$$rcXR;deUR(dS[U5URU'/URU'[ UR5UR :aSUlggUR (aUR HnURRURR:XdM3URUnX$:waI[UUURRURR5nURU5 gUR RU5nX`RU' g gg)aAdd the collected test items from a node. Collection is complete once all nodes have submitted their collection. In this case its pending list is set to an empty list. When the collection is already completed this submission is from a node which was restarted to replace a dead node. In this case we already assign the pending items here. In either case ``.schedule()`` will instruct the node to start running the required tests. TN) rrrrrrrgatewayspecridrpop)rr- collectiondeadnodedead_collectionmsgr%s radd_node_collection"EachScheduling.add_node_collectionIs(((((++)-j)9D  &&(D  d #4''(DMM9/3,:  " " 11##((DLL,=,==&*&:&:8&DO!44+&$,,// LLOO   "3377AG.5%%d+2#rc@URURU5 gr,)rremove)rr- item_indexdurations rmark_test_complete!EachScheduling.mark_test_completems $&&z2rc[5er,NotImplementedError)ritems rmark_test_pending EachScheduling.mark_test_pendingrs !##rc[5er,rB)rr-indicess rremove_pending_tests_from_node-EachScheduling.remove_pending_tests_from_nodeus "##rcURRU5nU(dgURUURS5nU(aX RU'U$)Nr)rr4rr)rr-r% crashitems r remove_nodeEachScheduling.remove_node|sR##''-((.w{{1~> *1 ! !$ 'rcUR(deURR5HupXR;aMU(dE[ [ UR U55USS&UR5 UR5 OURU5 URRU5 M g)a Schedule the test items on the nodes. If the node's pending list is empty it is a new node which needs to run all the tests. If the pending list is already populated (by ``.add_node_collection()``) then it replaces a dead node and we only need to run those tests. N) rritemsrrangerrsend_runtest_allshutdownsend_runtest_someappend)rr-r%s rscheduleEachScheduling.schedules++++!..446MD}}$"3t';';D'A#BC %%' &&w/ MM  &7r)rrrrrrrrr,)rz pytest.ConfigrzProducer | NonereturnNone)rXzlist[WorkerController])rXbool)r-rrXrY)r-rr5z Sequence[str]rXrY)r)r-rr=intr>floatrXrY)rDstrrXrY)r-rrHz Sequence[int]rXrY)r-rrXz str | None)rXrY)__name__ __module__ __qualname____firstlineno____doc__rpropertyrr&r)r.r9r?rErIrMrV__static_attributes__rrr r s  -..  %"$"2?" "JJK3$3253AF3 3 $$$$  $'rr ) __future__rcollections.abcrpytest xdist.remoter xdist.reportrxdist.workermanagerrr rerrrls&"$ !/3/K'K'r