From 8e4bd5ae1e223ad4d5f8cb2ba3d4e1a23309b183 Mon Sep 17 00:00:00 2001
From: Krishna Kumar <3963513+kks32@users.noreply.github.com>
Date: Fri, 4 Oct 2024 09:36:44 -0500
Subject: [PATCH] Add GNS builds on PRs as well

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

diff --git a/.github/workflows/train.yml b/.github/workflows/train.yml
index 366b9f1..b489f9f 100644
--- a/.github/workflows/train.yml
+++ b/.github/workflows/train.yml
@@ -2,6 +2,7 @@ name: GNS Train and Test
 
 on:
   push:
+  pull_request:
 
 jobs:
   gns:
@@ -26,4 +27,4 @@ jobs:
         TMP_DIR="../gns-sample"
         DATASET_NAME="WaterDropSample"
         git clone https://github.com/geoelements/gns-sample ../gns-sample
-        python -m gns.train mode="train" training.steps=10
\ No newline at end of file
+        python -m gns.train mode="train" training.steps=10