From 9a3866b785562cb61362687140b34f1853c94b19 Mon Sep 17 00:00:00 2001
From: Michael Paskett <michaelpaskett@meta.com>
Date: Wed, 22 Jan 2025 10:59:48 -0800
Subject: [PATCH] model training bugfix (#7754)

Summary:

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

Reviewed By: JacobSzwejbka

Differential Revision: D68233527
---
 extension/training/examples/XOR/export_model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/extension/training/examples/XOR/export_model.py b/extension/training/examples/XOR/export_model.py
index a245361e18b..bfbe0ce2138 100644
--- 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: