missing req

MadryLab · Aug 21, 2019 · 3612141 · 3612141
1 parent 8e14641
commit 3612141
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -45,7 +45,7 @@
 
 autodoc_mock_imports = ['torch', 'numpy', 'pandas', 'tensorboardX', 'dill',
         'torchvision', 'scipy', 'GPUtil', 'tables', 'scikit-learn', 'seaborn',
-        'cox', 'matplotlib']
+        'cox', 'matplotlib', 'sklearn']
 
 autoclass_content = "both"