Skip to content

Commit baf9f2d

Browse files
authored
chore(DAGs): Remove deprecated sft, rl, and gemma3-4b e2e DAGs (GoogleCloudPlatform#1303)
This change removes the following deprecated DAGs in accordance with b/528415968, b/528424941, and b/528418808: - `maxtext_sft.py` - `maxtext_rl.py` - `maxtext_sft_notebook.py` - `maxtext_rl_notebook.py` - Gemma3-4b from `maxtext_end_to_end.py`
1 parent 61c0816 commit baf9f2d

6 files changed

Lines changed: 0 additions & 583 deletions

File tree

dags/common/scheduling_helper/dag_integrity_whitelist.yaml

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,10 +75,6 @@ whitelisted_dags:
7575
- maxtext_regular_restore_with_node_disruption
7676
- maxtext_regular_restore_with_resumed_workload
7777
- maxtext_regular_save
78-
- maxtext_rl
79-
- maxtext_rl_notebook
80-
- maxtext_sft
81-
- maxtext_sft_notebook
8278
- maxtext_sft_trainer
8379
- maxtext_sweep_gke_example_dag
8480
- maxtext_trillium_configs_perf

dags/multipod/maxtext_end_to_end.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -69,14 +69,6 @@
6969
"bash tests/end_to_end/tpu/gemma2/2b/test_gemma2_to_hf.sh",
7070
],
7171
},
72-
"gemma3-4b": {
73-
"owner": test_owner.HENGTAO_G,
74-
"commands": [
75-
"export RUN_NAME=$(date +%Y-%m-%d-%H-%M-%S)",
76-
"bash tests/end_to_end/tpu/gemma3/4b/test_gemma3_to_mt.sh $RUN_NAME",
77-
"bash tests/end_to_end/tpu/gemma3/4b/test_gemma3.sh $RUN_NAME",
78-
],
79-
},
8072
"qwen3-4b": {
8173
"owner": test_owner.HENGTAO_G,
8274
"commands": [

dags/post_training/maxtext_rl.py

Lines changed: 0 additions & 189 deletions
This file was deleted.

dags/post_training/maxtext_rl_notebook.py

Lines changed: 0 additions & 100 deletions
This file was deleted.

0 commit comments

Comments
 (0)