From 84059dbda23796f9691a340826a7af8b12d7e90a Mon Sep 17 00:00:00 2001 From: Morgan Epp <60796713+epmog@users.noreply.github.com> Date: Mon, 29 Jun 2026 09:10:34 -0500 Subject: [PATCH] chore: rename YamlLoader to SafeLoader Signed-off-by: Morgan Epp <60796713+epmog@users.noreply.github.com> --- .../model_v0/benchmark/test_yaml_loader_performance.py | 8 ++++---- .../model_v1/benchmark/test_yaml_loader_performance.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/test/openjd/model_v0/benchmark/test_yaml_loader_performance.py b/test/openjd/model_v0/benchmark/test_yaml_loader_performance.py index 04d2b4b0..ce2cca47 100644 --- a/test/openjd/model_v0/benchmark/test_yaml_loader_performance.py +++ b/test/openjd/model_v0/benchmark/test_yaml_loader_performance.py @@ -164,17 +164,17 @@ def benchmark_loader( # Select the appropriate loader directly if loader_type == "CSafeLoader": try: - from yaml import CSafeLoader as YamlLoader # type: ignore[attr-defined] + from yaml import CSafeLoader as SafeLoader # type: ignore[attr-defined] except ImportError: - from yaml import SafeLoader as YamlLoader # type: ignore[assignment] + from yaml import SafeLoader # type: ignore[assignment] else: - from yaml import SafeLoader as YamlLoader # type: ignore[assignment] + from yaml import SafeLoader # type: ignore[assignment] for _ in range(iterations): start_time = time.perf_counter() # Parse YAML directly instead of using document_string_to_object # to avoid the module-level loader selection - yaml.load(template_content, Loader=YamlLoader) + yaml.load(template_content, Loader=SafeLoader) end_time = time.perf_counter() times.append((end_time - start_time) * 1000) # Convert to milliseconds diff --git a/test/openjd/model_v1/benchmark/test_yaml_loader_performance.py b/test/openjd/model_v1/benchmark/test_yaml_loader_performance.py index 04d2b4b0..ce2cca47 100644 --- a/test/openjd/model_v1/benchmark/test_yaml_loader_performance.py +++ b/test/openjd/model_v1/benchmark/test_yaml_loader_performance.py @@ -164,17 +164,17 @@ def benchmark_loader( # Select the appropriate loader directly if loader_type == "CSafeLoader": try: - from yaml import CSafeLoader as YamlLoader # type: ignore[attr-defined] + from yaml import CSafeLoader as SafeLoader # type: ignore[attr-defined] except ImportError: - from yaml import SafeLoader as YamlLoader # type: ignore[assignment] + from yaml import SafeLoader # type: ignore[assignment] else: - from yaml import SafeLoader as YamlLoader # type: ignore[assignment] + from yaml import SafeLoader # type: ignore[assignment] for _ in range(iterations): start_time = time.perf_counter() # Parse YAML directly instead of using document_string_to_object # to avoid the module-level loader selection - yaml.load(template_content, Loader=YamlLoader) + yaml.load(template_content, Loader=SafeLoader) end_time = time.perf_counter() times.append((end_time - start_time) * 1000) # Convert to milliseconds