From d108f742e61eaf37674d69336e3b080f92372cd6 Mon Sep 17 00:00:00 2001
From: Mark Saroufim <marksaroufim@meta.com>
Date: Mon, 18 Nov 2024 20:35:29 -0800
Subject: [PATCH] update

---
 .github/workflows/nvidia_workflow.yml | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/.github/workflows/nvidia_workflow.yml b/.github/workflows/nvidia_workflow.yml
index fdbe9fe..828b943 100644
--- a/.github/workflows/nvidia_workflow.yml
+++ b/.github/workflows/nvidia_workflow.yml
@@ -31,14 +31,10 @@ jobs:
 
     - name: Install dependencies
       run: |
-        if grep -rE "(import numpy|from numpy)" "${{ github.event.inputs.filename }}"; then
-          echo "Numpy detected, installing numpy"
-          pip install numpy
-        fi
         # Check if 'import torch' is in any Python file
         if grep -rE "(import torch|from torch)" "${{ github.event.inputs.filename }}"; then
           echo "PyTorch detected, installing torch"
-          pip install torch
+          pip install numpy torch
         fi
         # Check if 'import triton' is in any Python file
         if grep -rE "(import triton|from triton)" "${{ github.event.inputs.filename }}"; then