From ce87359509be993377f0fa7e7828c845310ff714 Mon Sep 17 00:00:00 2001
From: Krishna Kumar <krishnak@utexas.edu>
Date: Thu, 12 Dec 2024 09:51:31 +0530
Subject: [PATCH] Use v2 docker container in pytest

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

diff --git a/.github/workflows/train.yml b/.github/workflows/train.yml
index b489f9f..8d27ca9 100644
--- a/.github/workflows/train.yml
+++ b/.github/workflows/train.yml
@@ -8,7 +8,7 @@ jobs:
   gns:
     runs-on: ubuntu-latest
     container:
-      image: ghcr.io/geoelements/gns:dataloader
+      image: ghcr.io/geoelements/gns:v2
 
     steps:
     - name: Checkout repository