Skip to content

Commit 461ab55

Browse files
authored
Fix notebook file path for post-training DAGs (GoogleCloudPlatform#1171)
Fix notebook file path for post-training DAG
1 parent 21614ce commit 461ab55

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

dags/post_training/maxtext_rl_notebook.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@
8383
rl_notebook_test = notebook_util.initialize_notebook_test(
8484
test_name=f"{DAG_TEST_NAME}_rl_{loss_algo.value}",
8585
dag_name=DAG_TEST_NAME,
86-
notebook_path="src/MaxText/examples/rl_llama3_demo.ipynb",
86+
notebook_path="src/maxtext/examples/rl_llama3_demo.ipynb",
8787
set_up_script=setup_script,
8888
parameters={"LOSS_ALGO": loss_algo.loss_name},
8989
task_owner=test_owner.DEPP_L,

dags/post_training/maxtext_sft_notebook.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@
7575
sft_notebook_test = notebook_util.initialize_notebook_test(
7676
test_name=f"{DAG_TEST_NAME}_sft",
7777
dag_name=DAG_TEST_NAME,
78-
notebook_path="src/MaxText/examples/sft_llama3_demo.ipynb",
78+
notebook_path="src/maxtext/examples/sft_llama3_demo.ipynb",
7979
set_up_script=setup_script,
8080
parameters={},
8181
task_owner=test_owner.DEPP_L,

0 commit comments

Comments
 (0)