From acf7abbed27cc1392671a2fb6bb189093b7d9cef Mon Sep 17 00:00:00 2001
From: Aaron <29749331+aarnphm@users.noreply.github.com>
Date: Sun, 16 Jul 2023 19:52:06 -0400
Subject: [PATCH] chore: enable cache

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
---
 .github/workflows/ci.yml | 5 ++++-
 pyproject.toml           | 7 +++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index b1d8cb3d4c3..edb5fbcfa27 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -57,10 +57,13 @@ jobs:
       - name: Set up PDM
         uses: pdm-project/setup-pdm@v3
         with:
+          cache: true
           python-version: '3.8'
           enable-pep582: false
       - name: Install dependencies
-        run: pdm install -d -G testing -G frameworks
+        run: |
+          pdm use -f "3.8"
+          pdm install -d -G testing -G frameworks
       - name: Run framework integration tests
         run: pdm run -v integration
       - name: Disambiguate coverage filename
diff --git a/pyproject.toml b/pyproject.toml
index 36971f229f4..f67a02ad6d3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -227,8 +227,11 @@ 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 --force-reinstall torch --index-url https://download.pytorch.org/whl/cpu"
-integration = { composite = ["tests tests/integration -n auto"] }
+pre_integration = "pip install torch --index-url https://download.pytorch.org/whl/cpu"
+integration = { composite = [
+    "pip install -U tensorflow",
+    "tests tests/integration -n auto",
+] }
 monitoring = { composite = ["tests examples/monitoring/task_classification"] }
 e2e = { composite = [
     "tests tests/e2e/bento_server_http",