0 0 1 Python

modify tests to take an nn.Module (#1609)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/1609

as title, we are changing torch.export to require an nn.Module, so this fixes tests that are broken by that change.

Reviewed By: JacobSzwejbka

Differential Revision: D52809845

fbshipit-source-id: f0e015ccae79db02520e6edf6502d8e4475699b6

Michael Suo committed 2y ago

33f8887bb3149d606127352a251503e8e2f24f60

Parent: 41fe5d2

Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 1/18/2024, 1:55:40 AM