Fix test_modules_can_be_imported (#145387)

`test_modules_can_be_imported` test is currently failing due to a few missing private modules and this PR gets it working before I start to clean up the public allow list
Pull Request resolved: https://github.com/pytorch/pytorch/pull/145387
Approved by: https://github.com/albanD
This commit is contained in:
Howard Huang 2025-01-22 10:12:15 -08:00 committed by PyTorch MergeBot
parent 768ad0886f
commit bf4f8919df

View file

@ -308,6 +308,7 @@ class TestPublicBindings(TestCase):
"torch.onnx._internal.exporter._reporting",
"torch.onnx._internal.exporter._schemas",
"torch.onnx._internal.exporter._tensors",
"torch.onnx._internal.exporter._torchlib.ops",
"torch.onnx._internal.exporter._verification",
"torch.onnx._internal.fx._pass",
"torch.onnx._internal.fx.analysis",
@ -377,6 +378,7 @@ class TestPublicBindings(TestCase):
"torch.distributed._spmd.experimental_ops",
"torch.distributed._spmd.parallel_mode",
"torch.distributed._tensor",
"torch.distributed._tools.sac_ilp",
"torch.distributed.algorithms._checkpoint.checkpoint_wrapper",
"torch.distributed.algorithms._optimizer_overlap",
"torch.distributed.rpc._testing.faulty_agent_backend_registry",