diff --git a/.circleci/config.yml b/.circleci/config.yml index 13f387a7..52aee499 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -8,7 +8,7 @@ setup_venv: &setup_venv . ~/crypten-test/bin/activate pip3 install --upgrade pip pip3 install onnx==1.7.0 tensorboard pandas - pip3 install sklearn --use-pep517 + pip3 install scikit-learn --use-pep517 pip3 install torch>=1.7.0 torchvision>=0.8.1 pip3 install tensorflow pip3 install tf2onnx diff --git a/requirements.txt b/requirements.txt index 140fdf8d..47258683 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,4 +7,4 @@ pyyaml>=5.3.1 tensorboard future scipy>=1.6.0 -sklearn +scikit-learn