diff --git a/docs/source/conf.py b/docs/source/conf.py index c1e193917da2a..e9cfddcb33136 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -295,7 +295,7 @@ def setup(app): MOCK_REQUIRE_PACKAGES.append(pkg.rstrip()) # TODO: better parse from package since the import name and package name may differ -MOCK_MANUAL_PACKAGES = ['torch', 'torchvision', 'sklearn', 'test_tube', 'mlflow', 'comet_ml', 'wandb', 'neptune'] +MOCK_MANUAL_PACKAGES = ['torch', 'torchvision', 'test_tube', 'mlflow', 'comet_ml', 'wandb', 'neptune'] autodoc_mock_imports = MOCK_REQUIRE_PACKAGES + MOCK_MANUAL_PACKAGES # for mod_name in MOCK_REQUIRE_PACKAGES: # sys.modules[mod_name] = mock.Mock() diff --git a/requirements.txt b/requirements.txt index 92731a1a91788..23a965bf441a0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,3 @@ -scikit-learn>=0.20.2 tqdm>=4.35.0 numpy>=1.16.4 torch>=1.1