diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index 2f7c46f330..68ab10fa0c 100644 --- a/.github/workflows/self-push.yml +++ b/.github/workflows/self-push.yml @@ -35,7 +35,7 @@ jobs: - name: Install dependencies run: | source .env/bin/activate - pip install torch==1.4.0 + pip install torch pip install .[sklearn,testing] - name: Are GPUs recognized by our DL frameworks