From 6baf691210f43cf56e8ad6ba9ea9ae9078d548f7 Mon Sep 17 00:00:00 2001
From: brandon-b-miller <brmiller@nvidia.com>
Date: Wed, 15 Nov 2023 14:17:18 -0800
Subject: [PATCH] add to needs section

---
 .github/workflows/pr.yaml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
index c392cdd9..1a7d1eab 100644
--- a/.github/workflows/pr.yaml
+++ b/.github/workflows/pr.yaml
@@ -15,7 +15,8 @@ jobs:
     needs:
       - checks
       - wheel-build
-      - wheel-test
+      - wheel-test-amd64
+      - wheel-test-arm64
     secrets: inherit
     uses: rapidsai/shared-action-workflows/.github/workflows/pr-builder.yaml@branch-23.12
   checks: