[raysgd] Fix More Docs (#7565)

2026-07-06 01:07:38 +08:00 · 2020-03-11 14:17:47 -07:00
parent d046faeb9c
commit b38ed4be71
1 changed files with 1 additions and 0 deletions
@@ -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()