From 27f77c9e253b12f9dc4a63771c8e0cae8f6d6965 Mon Sep 17 00:00:00 2001
From: Aaron <29749331+aarnphm@users.noreply.github.com>
Date: Sun, 16 Jul 2023 22:16:57 -0400
Subject: [PATCH] fix: detectron tests

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 43552255932..dc55e724217 100644
--- 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",