make tests pass with enable_first_class_module() enabled. (#21565)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21565
ghimport-source-id: d1fe735fb7821eadc59116fb921d8fe39a49f818
Reviewed By: driazati
Differential Revision: D15729503
Pulled By: zdevito
fbshipit-source-id: fabb678f040d21fae7545e3b2be1d098e24c544e
