From 6052068d236c3ca11053903bcbc3971737722bac Mon Sep 17 00:00:00 2001 From: Jiri Borovec Date: Sat, 8 Feb 2020 00:21:35 +0100 Subject: [PATCH] drop sklearn dependency --- docs/source/conf.py | 2 +- requirements.txt | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/docs/source/conf.py b/docs/source/conf.py index a5d7841aa649c..c96dd3d245d5f 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 90cec5a5f5606..d4116cc6a597c 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