executorch/extension/training/pybindings/test (#7390)

Summary: Pull Request resolved: #7390

Reviewed By: avikchaudhuri, ydwu4

Differential Revision: D67383993

Author: gmagogsfm

extension/training/pybindings/test/test.py

@@ -33,7 +33,7 @@ def get_random_inputs(self):
 
     def test(self):
         m = self.ModuleSimpleTrain()
-        ep = torch.export.export(m, m.get_random_inputs())
+        ep = torch.export.export(m, m.get_random_inputs(), strict=True)
         ep = _export_forward_backward(ep)
         ep = to_edge(ep)
         ep = ep.to_executorch()