From b38ed4be71d4be096f324c8c666e01e8af5f6c8b Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Wed, 11 Mar 2020 14:17:47 -0700 Subject: [PATCH] [raysgd] Fix More Docs (#7565) --- doc/source/conf.py | 1 + 1 file changed, 1 insertion(+) diff --git a/doc/source/conf.py b/doc/source/conf.py index 964471ebc..24297e98b 100644 --- a/doc/source/conf.py +++ b/doc/source/conf.py @@ -51,6 +51,7 @@ MOCK_MODULES = [ "torch.nn", "torch.nn.parallel", "torch.utils.data", + "torch.utils.data.distributed" ] for mod_name in MOCK_MODULES: sys.modules[mod_name] = mock.Mock()