Fix bug where explicitly providing a namespace never worked. (#40830)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40830
Fixes #40725
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Test Plan: Imported from OSS
Differential Revision: D22323886
Pulled By: ezyang
fbshipit-source-id: b8a61496923d9f086d4c201024748505ba783238