From f511cd2b8a5066b2d6cc13c9c0a15bbe4f6987b8 Mon Sep 17 00:00:00 2001
From: Mohit Khatwani <118776932+khatwanimohit@users.noreply.github.com>
Date: Wed, 21 Feb 2024 11:49:14 -0800
Subject: [PATCH] Renamed test_gamma.sh to test_gemma.sh (#135)

---
 dags/multipod/maxtext_end_to_end.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dags/multipod/maxtext_end_to_end.py b/dags/multipod/maxtext_end_to_end.py
index 29af9638..8a4134ba 100644
--- a/dags/multipod/maxtext_end_to_end.py
+++ b/dags/multipod/maxtext_end_to_end.py
@@ -38,7 +38,7 @@
   test_models = {
       "llama2": ["test_llama2", "llama_finetuning_test"],
       "mistral": ["test_mistral"],
-      "gamma": ["test_gamma"],
+      "gemma": ["test_gemma"],
       "gpt3": ["test_gpt3"],
   }
   test_modes = [SetupMode.STABLE, SetupMode.NIGHTLY]