fix: detectron tests

Signed-off-by: Aaron <[email protected]>
bentoml · Jul 17, 2023 · 27f77c9 · 27f77c9
1 parent 421cea5
commit 27f77c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -228,7 +228,7 @@ dev = [
 _run_test = "pytest --cov=bentoml --cov-report=term-missing --cov-config=pyproject.toml -vv"
 tests = { composite = ["_run_test {args:tests}"] }
 unit = { composite = ["tests tests/unit -n 3"] }
-pre_integration = "pip install torch --index-url https://download.pytorch.org/whl/cpu"
+pre_integration = "pip install torch torchaudio torchvision --index-url https://download.pytorch.org/whl/cpu"
 integration = { composite = [
     "pip install -U tensorflow onnx onnxruntime",
     "pip install git+https://github.com/facebookresearch/detectron2.git#egg=detectron2",