@@ -673,6 +673,7 @@ def task_path_decider(check_file_exists: bool = False) -> str:
673673
674674 return group
675675
676+
676677@dataclasses .dataclass
677678class XpkNameGenAndQuarantineTask (XpkTask ):
678679 """Task for running XPK workloads with name generation and quarantine."""
@@ -683,8 +684,12 @@ class XpkNameGenAndQuarantineTask(XpkTask):
683684
684685 # Temporary variables to pass tasks from run() to intermediary_flow()
685686 _run_name_task : Any = dataclasses .field (default = None , init = False , repr = False )
686- _tb_file_location_task : Any = dataclasses .field (default = None , init = False , repr = False )
687- _profile_file_location_task : Any = dataclasses .field (default = None , init = False , repr = False )
687+ _tb_file_location_task : Any = dataclasses .field (
688+ default = None , init = False , repr = False
689+ )
690+ _profile_file_location_task : Any = dataclasses .field (
691+ default = None , init = False , repr = False
692+ )
688693
689694 def run (
690695 self ,
@@ -724,10 +729,14 @@ def run(
724729
725730 # Update profile file location if profile config exists
726731 if self .task_metric_config .profile :
727- self ._profile_file_location_task = name_format .generate_profile_file_location (
728- self ._run_name_task , self .task_metric_config .profile .file_location
732+ self ._profile_file_location_task = (
733+ name_format .generate_profile_file_location (
734+ self ._run_name_task , self .task_metric_config .profile .file_location
735+ )
736+ )
737+ self .task_metric_config .profile .file_location = (
738+ self ._profile_file_location_task
729739 )
730- self .task_metric_config .profile .file_location = self ._profile_file_location_task
731740
732741 # 2. Run under quarantine task group if applicable
733742 if QuarantineTests .is_quarantined (test_name ) and self .quarantine_task_group :
@@ -770,15 +779,11 @@ def intermediary_flow(
770779 )
771780 else :
772781 flow = (
773- self ._run_name_task
774- >> self ._tb_file_location_task
775- >> launch_workload
782+ self ._run_name_task >> self ._tb_file_location_task >> launch_workload
776783 )
777784 return flow
778785
779786
780-
781-
782787@dataclasses .dataclass
783788class GpuCreateResourceTask (BaseTask ):
784789 """This is a class to set up tasks for GPU.
0 commit comments