From 5bdb0258d693d0faefc768b7288d7c3149a9eb58 Mon Sep 17 00:00:00 2001
From: Ean Garvey <87458719+monorimet@users.noreply.github.com>
Date: Mon, 15 Jul 2024 11:53:18 -0500
Subject: [PATCH] Explicitly install torch cpu in SHARK CI

---
 .github/workflows/test_shark.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/test_shark.yml b/.github/workflows/test_shark.yml
index 6f2e4b4e..45ada02d 100644
--- a/.github/workflows/test_shark.yml
+++ b/.github/workflows/test_shark.yml
@@ -49,6 +49,7 @@ jobs:
           cd $GITHUB_WORKSPACE/SHARK
           python${{ matrix.version }} -m venv shark.venv
           source shark.venv/bin/activate
+          pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
           pip install -r requirements.txt --no-cache-dir
           pip install -e .
           python apps/shark_studio/tests/api_test.py