Fix train_rl and rl.yml path

Author: Depp Lee

dags/post_training/maxtext_rl.py

@@ -79,7 +79,7 @@
           f"{test_config_util.DEFAULT_BUCKET}/llama3.1-70b-Instruct/"
           "scanned-pathways/0/items/"
       ),
-      rl_config_path="src/MaxText/configs/rl.yml",
+      rl_config_path="src/maxtext/configs/post_train/rl.yml",
       loss_algos=[
           test_config_util.LossAlgo.GRPO,
           test_config_util.LossAlgo.GSPO,

dags/post_training/util/test_config_util.py

@@ -88,7 +88,7 @@ class RLTestConfig:
   tokenizer_path: str
   load_parameters_path: str
   loss_algos: list[LossAlgo]
-  rl_config_path: str = "src/MaxText/configs/rl.yml"
+  rl_config_path: str = "src/maxtext/configs/post_train/rl.yml"
 
   def __init__(
       self,
@@ -100,7 +100,7 @@ def __init__(
       tokenizer_path: str,
       load_parameters_path: str,
       loss_algos: list[LossAlgo],
-      rl_config_path: str = "src/MaxText/configs/rl.yml",
+      rl_config_path: str = "src/maxtext/configs/post_train/rl.yml",
   ):
     """Initializes the RL test configurations.
 
@@ -158,7 +158,7 @@ def generate_rl_training_command(
       ])
 
     rl_command = (
-        "python -m src.MaxText.rl.train_rl "
+        "python -m src.maxtext.rl.train_rl "
         f"{self.rl_config_path} " + " ".join(command_params)
     )