Adding scikit-learn for the tests

Author: nicolas-chaulet

.github/workflows/tests.yaml

@@ -14,7 +14,7 @@ jobs:
             - name: Install dependencies
               run: |
                   python -m pip install --upgrade pip
-                  pip install torch numpy flake8 setuptools
+                  pip install torch numpy scikit-learn flake8 setuptools
             - name: Build package
               run: |
                   python setup.py build_ext --inplace