Fix YAML Loader error

triton-inference-server · Jun 18, 2021 · 972bb9b · 972bb9b
1 parent f3d0961
commit 972bb9b
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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)