Update tests with aiu arguments

Signed-off-by: Andrea Fasoli <[email protected]>

Author: andrea-fasoli

tests/test_run_quant.py

@@ -110,6 +110,7 @@ def test_parse_arguments(job_config):
         _,
         _,
         _,
+        _,
     ) = parse_arguments(parser, job_config_copy)
     assert str(model_args.torch_dtype) == "torch.bfloat16"
     assert data_args.training_data_path == "data_train"
@@ -132,6 +133,7 @@ def test_parse_arguments_defaults(job_config):
         fms_mo_args,
         _,
         _,
+        _,
     ) = parse_arguments(parser, job_config_defaults)
     assert str(model_args.torch_dtype) == "torch.bfloat16"
     assert model_args.model_revision == "main"