Paramter parsing bugfix

NVIDIA · Jun 6, 2024 · b0a4dd3 · b0a4dd3
1 parent be70812
commit b0a4dd3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/export/test_nemo_export.py b/tests/export/test_nemo_export.py
@@ -515,8 +515,7 @@ def run_inference_tests(args):
     else:
         args.run_accuracy = False
 
-    if args.use_custom_all_reduce == "True":
-        args.use_custom_all_reduce = True
+    args.use_custom_all_reduce = args.use_custom_all_reduce == "True"
 
     if args.run_accuracy:
         if args.test_data_path is None: