Fix .module() calls (#2258)

Summary:

D53827930

Reviewed By: aakhundov

Differential Revision: D54542173

Author: angelayi

exir/emit/test/test_emit.py

@@ -215,7 +215,7 @@ def forward(self, x):
                 return [((1, 3, 1.2), True, [x + x, x * x])]
 
         ep = torch.export.export(M(), (torch.ones(2, 3),))
-        res = ep(torch.ones(2, 3))
+        res = ep.module()(torch.ones(2, 3))
         self.assertEqual(res[0][0], (1, 3, 1.2))
         program = to_edge(ep).to_executorch().executorch_program
         outputs = program.execution_plan[0].outputs
@@ -235,7 +235,7 @@ def forward(self, x, y, z):
                 return x + y, x + x, x + y + z
 
         ep = torch.export.export(M(), (torch.ones(2, 3), 2, True))
-        ep(torch.ones(2, 3), 2, True)
+        ep.module()(torch.ones(2, 3), 2, True)
         program = to_edge(ep).to_executorch().executorch_program
         inputs = program.execution_plan[0].inputs
         self.assertEqual(len(inputs), 3)