Skip to content

Commit e719758

Browse files
committed
Fix pre-commit errors
1 parent 402b03e commit e719758

1 file changed

Lines changed: 15 additions & 10 deletions

File tree

xlml/apis/task.py

Lines changed: 15 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -673,6 +673,7 @@ def task_path_decider(check_file_exists: bool = False) -> str:
673673

674674
return group
675675

676+
676677
@dataclasses.dataclass
677678
class 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
783788
class GpuCreateResourceTask(BaseTask):
784789
"""This is a class to set up tasks for GPU.

0 commit comments

Comments
 (0)