@@ -317,7 +317,7 @@ class _ReplicaAnalysis:
317317
318318
319319@dataclass
320- class _ActiveRunAnalysis :
320+ class _RunAnalysis :
321321 """Aggregated replica analysis used to determine the run's next status.
322322
323323 Each replica contributes `RunStatus` based on its jobs' statuses.
@@ -379,8 +379,8 @@ async def _analyze_active_run(
379379 run : Run ,
380380 run_jobs_by_position : Dict [Tuple [int , int ], Job ],
381381 retry_single_job : bool ,
382- ) -> _ActiveRunAnalysis :
383- analysis = _ActiveRunAnalysis ()
382+ ) -> _RunAnalysis :
383+ run_analysis = _RunAnalysis ()
384384 for replica_num , job_models in group_jobs_by_replica_latest (run_model .jobs ):
385385 replica_analysis = await _analyze_active_run_replica (
386386 session = session ,
@@ -390,8 +390,8 @@ async def _analyze_active_run(
390390 replica_num = replica_num ,
391391 job_models = job_models ,
392392 )
393- _update_active_run_analysis ( analysis , replica_analysis , retry_single_job )
394- return analysis
393+ _apply_replica_analysis ( run_analysis , replica_analysis , retry_single_job )
394+ return run_analysis
395395
396396
397397async def _analyze_active_run_replica (
@@ -455,8 +455,8 @@ async def _analyze_active_run_replica(
455455 )
456456
457457
458- def _update_active_run_analysis (
459- analysis : _ActiveRunAnalysis ,
458+ def _apply_replica_analysis (
459+ analysis : _RunAnalysis ,
460460 replica_analysis : _ReplicaAnalysis ,
461461 retry_single_job : bool ,
462462) -> None :
@@ -519,7 +519,7 @@ def _job_needs_retry_evaluation(job_model: JobModel) -> bool:
519519 )
520520
521521
522- def _get_active_run_transition (run : Run , analysis : _ActiveRunAnalysis ) -> _ActiveRunTransition :
522+ def _get_active_run_transition (run : Run , analysis : _RunAnalysis ) -> _ActiveRunTransition :
523523 # Check `analysis.contributed_statuses` in the priority order.
524524 if RunStatus .FAILED in analysis .contributed_statuses :
525525 if RunTerminationReason .JOB_FAILED in analysis .termination_reasons :
0 commit comments