From b2611b7dfadf9f4ef0915523282bf1ae06da6450 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Wed, 2 Dec 2020 16:22:04 +0100 Subject: [PATCH] drop sklearn dependency (#4912) * drop sklearn dependency * scipy Co-authored-by: chaton Co-authored-by: Rohit Gupta --- requirements/extra.txt | 2 -- requirements/test.txt | 2 ++ 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements/extra.txt b/requirements/extra.txt index 2fd1e94a91..3d51d044e4 100644 --- a/requirements/extra.txt +++ b/requirements/extra.txt @@ -3,8 +3,6 @@ matplotlib>=3.1.1 horovod>=0.20.2 # no need to install with [pytorch] as pytorch is already installed omegaconf>=2.0.1 -# scipy>=0.13.3 -scikit-learn>=0.22.2 torchtext>=0.3.1, <0.7 # TODO: temporary fix fix for compatibility onnx>=1.7.0 onnxruntime>=1.3.0 diff --git a/requirements/test.txt b/requirements/test.txt index d553bb9b78..74c62f82c7 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -7,6 +7,8 @@ flake8>=3.6 flake8-black check-manifest twine==1.13.0 +# scipy>=0.13.3 +scikit-learn>=0.22.2 scikit-image>=0.17.2 black>=20.8b1 isort>=5.6.4