fix test_parallel_context.py

FlagOpen · Jan 2, 2025 · 79b7f12 · 79b7f12
1 parent c1e6890
commit 79b7f12
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit_tests/test_parallel_context.py b/tests/unit_tests/test_parallel_context.py
@@ -2,7 +2,7 @@
 
 from tests.unit_tests.test_utilities import Utils as MegatronUtils
 
-from megatron.training.arguments import parse_args
+from megatron.training.arguments import parse_args, validate_args
 import megatron.training.global_vars as mcore_global_vars
 from megatron.training.tokenizer.tokenizer import _NullTokenizer
 
@@ -49,6 +49,7 @@ def init_parallel_context() -> ParallelContext:
 
     train_args = FSTrainArguments(args)
     train_args.pre_validate_args()
+    validate_args(args)
     train_args.post_validate_args()
 
     # for building datasets