diff --git a/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_real_rollout.py b/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_real_rollout.py index 0a307d724c..21f42fc543 100644 --- a/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_real_rollout.py +++ b/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_real_rollout.py @@ -6,9 +6,7 @@ from tests.ci.ci_register import register_cuda_ci from tests.e2e.ft.conftest_ft.scenario_ft_random import run_ci -register_cuda_ci( - est_time=2400, suite="stage-c-8-gpu-h200", labels=["ft-long"], disabled="FT soak tests pending CI infra support" -) +register_cuda_ci(est_time=2400, suite="stage-c-8-gpu-h200", labels=["ft-long"]) _MODE: str = "dp2_cp2_real_rollout" diff --git a/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_tp2_ep2.py b/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_tp2_ep2.py index 2a9d0e7827..e1a46fc8bf 100644 --- a/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_tp2_ep2.py +++ b/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_tp2_ep2.py @@ -6,9 +6,7 @@ from tests.ci.ci_register import register_cuda_ci from tests.e2e.ft.conftest_ft.scenario_ft_random import run_ci -register_cuda_ci( - est_time=1800, suite="stage-c-8-gpu-h200", labels=["ft-long"], disabled="FT soak tests pending CI infra support" -) +register_cuda_ci(est_time=1800, suite="stage-c-8-gpu-h200", labels=["ft-long"]) _MODE: str = "dp2_cp2_tp2_ep2"