model training bugfix (#7754)

michaelpaskett-meta · facebook-github-bot · commit 9a3866b78556 · 2025-01-22T10:59:48.000-08:00
Summary:

Fails without return - no model returned from `_export_model()`

Reviewed By: JacobSzwejbka

Differential Revision: D68233527
diff --git a/extension/training/examples/XOR/export_model.py b/extension/training/examples/XOR/export_model.py
@@ -31,6 +31,7 @@ def _export_model():
     ep = to_edge(ep)
     # Lower the graph to executorch.
     ep = ep.to_executorch()
+    return ep
 
 
 def main() -> None:
