Skip to content

Commit 972bb9b

Browse files
Ashwin Rameshdzier
authored andcommitted
Fix YAML Loader error
1 parent f3d0961 commit 972bb9b

File tree

7 files changed

+7
-7
lines changed

7 files changed

+7
-7
lines changed

model_analyzer/config/input/config_command.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def _load_config_file(self, file_path):
6868
"""
6969

7070
with open(file_path, 'r') as config_file:
71-
config = yaml.safe_load(config_file, Loader=yaml.FullLoader)
71+
config = yaml.safe_load(config_file)
7272
return config
7373

7474
def set_config_values(self, args):

qa/L0_config_range/config_generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def get_all_configurations():
115115
with open(f'./config-{i}.yml', 'w') as file:
116116
yaml.dump(configuration, file)
117117
with open(f'./config-{i}.yml', 'r') as file:
118-
config = yaml.safe_load(file, Loader=yaml.FullLoader)
118+
config = yaml.safe_load(file)
119119
with open(f'./config-{i}.txt', 'w') as file:
120120
file.write(str(total_param))
121121

qa/L0_custom_flags/check_results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ def check_triton_per_model(self):
232232
args = parser.parse_args()
233233

234234
with open(args.config_file, 'r') as f:
235-
config = yaml.safe_load(f, Loader=yaml.FullLoader)
235+
config = yaml.safe_load(f)
236236

237237
TestOutputValidator(config, args.profile_models, args.analyzer_log_file,
238238
args.triton_log_file)

qa/L0_perf_analyzer/check_results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def check_time_window_50(self):
7171
args = parser.parse_args()
7272

7373
with open(args.config_file, 'r') as f:
74-
config = yaml.safe_load(f, Loader=yaml.FullLoader)
74+
config = yaml.safe_load(f)
7575

7676
TestOutputValidator(config, args.config_file, args.analyzer_log,
7777
args.test_name)

qa/L0_profile/check_results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,6 @@ def check_profile_logs(self):
8686
args = parser.parse_args()
8787

8888
with open(args.config_file, 'r') as f:
89-
config = yaml.safe_load(f, Loader=yaml.FullLoader)
89+
config = yaml.safe_load(f)
9090

9191
TestOutputValidator(config, args.test_name, args.analyzer_log_file)

qa/L0_results/check_results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,6 @@ def check_detailed_reports(self):
106106
args = parser.parse_args()
107107

108108
with open(args.config_file, 'r') as f:
109-
config = yaml.safe_load(f, Loader=yaml.FullLoader)
109+
config = yaml.safe_load(f)
110110

111111
TestOutputValidator(config, args.test_name, args.export_path)

qa/L0_state_management/check_results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ def check_measurements_consistent_with_config(self):
180180
args = parser.parse_args()
181181

182182
with open(args.config_file, 'r') as f:
183-
config = yaml.safe_load(f, Loader=yaml.FullLoader)
183+
config = yaml.safe_load(f)
184184

185185
TestOutputValidator(config, args.test_name, args.checkpoint_dir,
186186
args.analyzer_log_file)

0 commit comments

Comments
 (0)