Commit Graph

3 Commits

Author SHA1 Message Date
Michael Suo
ed4cda0183 [pkg] opt into autoformat
Summary: woooo

Test Plan: arc lint --apply-patches --take BLACK --paths-cmd 'hg files -I "caffe2/**/*.py"'

Reviewed By: SplitInfinity

Differential Revision: D28608934

fbshipit-source-id: 7768fed50a87883a95319376c0a6d73a9492bdcc
2021-05-21 15:03:52 -07:00
Lillian Johnson
76d2cb3b8e [torch.package/TorchScript] flag to gate allowance of TS serializaiton in torch.package (#57678)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57678

Test Plan: Imported from OSS

Reviewed By: suo

Differential Revision: D28232891

Pulled By: Lilyjjo

fbshipit-source-id: f6b2f4557cb98c4e811b7e3b665e0ffe88115555
2021-05-14 08:21:46 -07:00
Michael Suo
741d0f41d6 [package] split tests (#53749)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53749

Split up tests into cases that cover specific functionality. Goals:
1. Avoid the omnibus test file mess (see: test_jit.py) by imposing early
   structure and deliberately avoiding a generic TestPackage test case.
2. Encourage testing of individual APIs and components by example.
3. Hide the fake modules we created for these tests in their own folder.

You can either run the test files individually, or still use
test/test_package.py like before.

Also this isort + black formats all the tests.

Test Plan: Imported from OSS

Reviewed By: SplitInfinity

Differential Revision: D26958535

Pulled By: suo

fbshipit-source-id: 8a63048b95ca71f4f1aa94e53c48442686076034
2021-03-10 16:07:36 -08:00