stages_selected vs selected_stages

onnx · Dec 6, 2023 · 00d96f1 · 00d96f1
1 parent d85c9de
commit 00d96f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/cli.py b/test/cli.py
@@ -996,15 +996,15 @@ def test_026_cli_report(self):
         result_dict = report.get_dict(
             summary_csv_path,
             [
-                "stages_selected",
+                "selected_stages",
                 "stage_duration: export_pytorch",
                 "stage_state: export_pytorch",
             ],
         )
         for result in result_dict.values():
             # All of the models should have exported to ONNX, so the "onnx_exported" value
             # should be True for all of them
-            assert "export_pytorch" in yaml.safe_load(result["stages_selected"])
+            assert "export_pytorch" in yaml.safe_load(result["selected_stages"])
             assert yaml.safe_load(result["stage_state: export_pytorch"]) == "COMPLETED"
             assert yaml.safe_load(result["stage_duration: export_pytorch"]) > 0