more mock modules for docs

BasisResearch · Nov 10, 2021 · dcf5498 · dcf5498
1 parent 2f56fc7
commit dcf5498
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -74,7 +74,9 @@
 #     'exclude-members': '__dict__,__module__,__weakref__',
 # }
 
-MOCK_MODULES = ['numpy', 'pandas', 'torch', 'scipy', 'polyagamma']
+MOCK_MODULES = ['numpy', 'pandas', 'torch', 'polyagamma', 'torch.distributions',
+                'torch.linalg', 'torch.nn.functional']
+
 for mod_name in MOCK_MODULES:
    sys.modules[mod_name] = mock.Mock()