From f50eb4e7aa4b1f5d08652318abf4f57d459f55fd Mon Sep 17 00:00:00 2001
From: Julio Perez <jperez@nvidia.com>
Date: Sat, 14 Oct 2023 10:32:47 -0400
Subject: [PATCH] adding correct address for container

---
 .github/workflows/gpu.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/gpu.yml b/.github/workflows/gpu.yml
index ec080a40a2..a763013af3 100644
--- a/.github/workflows/gpu.yml
+++ b/.github/workflows/gpu.yml
@@ -13,7 +13,7 @@ jobs:
   gpu-ci-tensorflow:
     runs-on: linux-amd64-gpu-p100-latest-1
     container:
-      image: nvcr.io/nvidia/merlin/merlin-ci-runner:latest
+      image: nvcr.io/nvstaging/merlin/merlin-ci-runner:latest
       env:
         NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }}
       options: --shm-size=1G