Add missing module mocks in docs config

plstcharles · Jan 25, 2019 · c1d07e4 · c1d07e4
1 parent de642f5
commit c1d07e4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/docs/src/conf.py b/docs/src/conf.py
@@ -204,6 +204,7 @@
 
 if on_rtd:
     autodoc_mock_imports = [
+        'albumentations',
         'argparse',
         'augmentor',
         'cv2',
@@ -215,6 +216,7 @@
         'pip',
         'pynput',
         'pynput.keyboard',
+        'shapely',
         'sklearn',
         'sklearn.metrics',
         'tensorboardX',
@@ -229,6 +231,7 @@
         'torchvision',
         'torchvision.transforms',
         'torchvision.utils',
+        'tqdm',
     ]