test

alibaba · Oct 24, 2024 · 0b4cad0 · 0b4cad0
1 parent ff89eeb
commit 0b4cad0
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 9 deletions.
diff --git a/.github/workflows/unit_test.yml b/.github/workflows/unit_test.yml
@@ -28,8 +28,6 @@ jobs:
 
     - name: Checkout code
       uses: actions/checkout@v4
-      with:
-          ref: ${{ github.event.pull_request.head.sha }}
 
     - name: Run unit test
       run: |

diff --git a/chatlearn/__init__.py b/chatlearn/__init__.py
@@ -77,9 +77,3 @@ class RLHFMegatronModule(MegatronModule):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         logger.warning("RLHFMegatronModule is deprecated, please use MegatronModule")
-
-
-
-
-
-
diff --git a/tests/run_tests.sh b/tests/run_tests.sh
@@ -62,8 +62,8 @@ function run_test {
 
 
 function run_all_tests {
-  run_test python test_rlhf_data_input.py -c "configs/exp.yaml"
   run_test python test_data_dp.py -c "configs/rlhf.yaml"
+  run_test python test_rlhf_data_input.py -c "configs/exp.yaml"
   run_test python test_data_dp_zero.py -c "configs/rlhf.yaml"
   run_test python test_rlhf_colocate_forward_train.py -c "configs/rlhf2.yaml"
   run_test python test_evaluator_multi.py -c "configs/test_eval2.yaml"