From a78104296b6fe9bb000f89068b567f9d1b4c5803 Mon Sep 17 00:00:00 2001
From: Ben Frederickson <ben@benfrederickson.com>
Date: Mon, 4 Mar 2024 09:48:51 -0800
Subject: [PATCH] .

---
 .github/workflows/build.yaml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index e4fb94545..61ebdf0bc 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -35,11 +35,14 @@ jobs:
       date: ${{ inputs.date }}
       sha: ${{ inputs.sha }}
   rust-build:
-  #  needs: cpp-build
+    needs: cpp-build
     secrets: inherit
     uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@branch-24.02
     with:
+      build_type: ${{ inputs.build_type || 'branch' }}
+      branch: ${{ inputs.branch }}
       arch: "amd64"
+      date: ${{ inputs.date }}
       container_image: "rapidsai/ci-conda:latest"
       node_type: "gpu-v100-latest-1"
       run_script: "ci/build_rust.sh"