From 972bb9ba9ba7b0629ba2a663077c035f2ea64d10 Mon Sep 17 00:00:00 2001 From: Ashwin Ramesh Date: Fri, 18 Jun 2021 07:57:51 -0700 Subject: [PATCH] Fix YAML Loader error --- model_analyzer/config/input/config_command.py | 2 +- qa/L0_config_range/config_generator.py | 2 +- qa/L0_custom_flags/check_results.py | 2 +- qa/L0_perf_analyzer/check_results.py | 2 +- qa/L0_profile/check_results.py | 2 +- qa/L0_results/check_results.py | 2 +- qa/L0_state_management/check_results.py | 2 +- 7 files changed, 7 insertions(+), 7 deletions(-) diff --git a/model_analyzer/config/input/config_command.py b/model_analyzer/config/input/config_command.py index 149f6e621..dc3bcb88a 100644 --- a/model_analyzer/config/input/config_command.py +++ b/model_analyzer/config/input/config_command.py @@ -68,7 +68,7 @@ def _load_config_file(self, file_path): """ with open(file_path, 'r') as config_file: - config = yaml.safe_load(config_file, Loader=yaml.FullLoader) + config = yaml.safe_load(config_file) return config def set_config_values(self, args): diff --git a/qa/L0_config_range/config_generator.py b/qa/L0_config_range/config_generator.py index 27eae2ad0..ea8b4f970 100644 --- a/qa/L0_config_range/config_generator.py +++ b/qa/L0_config_range/config_generator.py @@ -115,7 +115,7 @@ def get_all_configurations(): with open(f'./config-{i}.yml', 'w') as file: yaml.dump(configuration, file) with open(f'./config-{i}.yml', 'r') as file: - config = yaml.safe_load(file, Loader=yaml.FullLoader) + config = yaml.safe_load(file) with open(f'./config-{i}.txt', 'w') as file: file.write(str(total_param)) diff --git a/qa/L0_custom_flags/check_results.py b/qa/L0_custom_flags/check_results.py index 398fca178..b9ca061ca 100644 --- a/qa/L0_custom_flags/check_results.py +++ b/qa/L0_custom_flags/check_results.py @@ -232,7 +232,7 @@ def check_triton_per_model(self): args = parser.parse_args() with open(args.config_file, 'r') as f: - config = yaml.safe_load(f, Loader=yaml.FullLoader) + config = yaml.safe_load(f) TestOutputValidator(config, args.profile_models, args.analyzer_log_file, args.triton_log_file) diff --git a/qa/L0_perf_analyzer/check_results.py b/qa/L0_perf_analyzer/check_results.py index 590347a22..664fe6a15 100644 --- a/qa/L0_perf_analyzer/check_results.py +++ b/qa/L0_perf_analyzer/check_results.py @@ -71,7 +71,7 @@ def check_time_window_50(self): args = parser.parse_args() with open(args.config_file, 'r') as f: - config = yaml.safe_load(f, Loader=yaml.FullLoader) + config = yaml.safe_load(f) TestOutputValidator(config, args.config_file, args.analyzer_log, args.test_name) diff --git a/qa/L0_profile/check_results.py b/qa/L0_profile/check_results.py index 5080763b0..b390c460c 100644 --- a/qa/L0_profile/check_results.py +++ b/qa/L0_profile/check_results.py @@ -86,6 +86,6 @@ def check_profile_logs(self): args = parser.parse_args() with open(args.config_file, 'r') as f: - config = yaml.safe_load(f, Loader=yaml.FullLoader) + config = yaml.safe_load(f) TestOutputValidator(config, args.test_name, args.analyzer_log_file) diff --git a/qa/L0_results/check_results.py b/qa/L0_results/check_results.py index 8a93b76d7..680ac1f9f 100644 --- a/qa/L0_results/check_results.py +++ b/qa/L0_results/check_results.py @@ -106,6 +106,6 @@ def check_detailed_reports(self): args = parser.parse_args() with open(args.config_file, 'r') as f: - config = yaml.safe_load(f, Loader=yaml.FullLoader) + config = yaml.safe_load(f) TestOutputValidator(config, args.test_name, args.export_path) diff --git a/qa/L0_state_management/check_results.py b/qa/L0_state_management/check_results.py index fee97f904..5c235a28a 100644 --- a/qa/L0_state_management/check_results.py +++ b/qa/L0_state_management/check_results.py @@ -180,7 +180,7 @@ def check_measurements_consistent_with_config(self): args = parser.parse_args() with open(args.config_file, 'r') as f: - config = yaml.safe_load(f, Loader=yaml.FullLoader) + config = yaml.safe_load(f) TestOutputValidator(config, args.test_name, args.checkpoint_dir, args.analyzer_log_file)