Fix ONNX test_quantize unittest (#6716)

This commit is contained in:
Funtowicz Morgan
2020-08-25 19:24:40 +02:00
committed by GitHub
parent 074340339a
commit ac9702c284
4 changed files with 22 additions and 24 deletions

View File

@@ -45,7 +45,7 @@ jobs:
source .env/bin/activate
pip install --upgrade pip
pip install torch!=1.6.0
pip install .[sklearn,testing]
pip install .[sklearn,testing,onnxruntime]
- name: Are GPUs recognized by our DL frameworks
run: |

View File

@@ -42,7 +42,7 @@ jobs:
source .env/bin/activate
pip install --upgrade pip
pip install torch!=1.6.0
pip install .[sklearn,testing]
pip install .[sklearn,testing,onnxruntime]
- name: Are GPUs recognized by our DL frameworks
run: |