lint

pytorch · Jun 5, 2023 · 93b0e16 · 93b0e16
1 parent dfbf214
commit 93b0e16
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/_export/__init__.py b/torch/_export/__init__.py
@@ -203,6 +203,7 @@ def export(
                 export_graph_signature,
                 flat_args,
             )
+            assert orig_out_spec is not None
             exported_program = ExportedProgram(
                 gm,
                 gm.graph,