diff --git a/docs/source/conf.py b/docs/source/conf.py index c1e193917d..e9cfddcb33 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -295,7 +295,7 @@ with open(os.path.join(PATH_ROOT, 'requirements.txt'), 'r') as fp: 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 92731a1a91..23a965bf44 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