Update on "make et.export support etrecord generation"

Gasoonjia · Gasoonjia · commit ab0983fed9f4 · 2025-08-07T11:50:46.000-07:00
this diff makes et.export etrecord generation supportive. Details can be found in #12925. After this change, all things in #12925 has completed. Differential Revision: [D79741917](https://our.internmc.facebook.com/intern/diff/D79741917/) [ghstack-poisoned]
diff --git a/export/tests/test_export_session.py b/export/tests/test_export_session.py
@@ -184,6 +184,7 @@ def test_context_propagation_through_pipeline(self) -> None:
             "export_recipe",
             "session_name",
             "artifact_dir",
+            "generate_etrecord"
         }
         self.assertEqual(set(session._run_context.keys()), expected_context_keys)
         self.assertEqual(session._run_context["session_name"], "test_session")
diff --git a/export/tests/test_export_stages.py b/export/tests/test_export_stages.py
@@ -307,6 +307,7 @@ def test_run_success(self, mock_to_edge: Mock) -> None:
             self.exported_programs,
             constant_methods=None,
             compile_config=mock_config,
+            generate_etrecord=False,
         )
 
         # Verify artifacts are set correctly

Original file line number	Diff line number	Diff line change
`@@ -184,6 +184,7 @@ def test_context_propagation_through_pipeline(self) -> None:`
`184`	`184`	`"export_recipe",`
`185`	`185`	`"session_name",`
`186`	`186`	`"artifact_dir",`
	`187`	`+ "generate_etrecord"`
`187`	`188`	`}`
`188`	`189`	`self.assertEqual(set(session._run_context.keys()), expected_context_keys)`
`189`	`190`	`self.assertEqual(session._run_context["session_name"], "test_session")`
Original file line number	Diff line number	Diff line change
`@@ -307,6 +307,7 @@ def test_run_success(self, mock_to_edge: Mock) -> None:`
`307`	`307`	`self.exported_programs,`
`308`	`308`	`constant_methods=None,`
`309`	`309`	`compile_config=mock_config,`
	`310`	`+ generate_etrecord=False,`
`310`	`311`	`)`
`311`	`312`
`312`	`313`	`# Verify artifacts are set correctly`