fix tags

GoogleCloudPlatform · Feb 4, 2025 · 3d429da · 3d429da
1 parent 07ee739
commit 3d429da
Show file tree

Hide file tree

Showing 6 changed files with 19 additions and 24 deletions.
diff --git a/dags/map_reproducibility/a3mega_gpt3_175b_nemo.py b/dags/map_reproducibility/a3mega_gpt3_175b_nemo.py
@@ -165,12 +165,11 @@ def run_aotc_workload():
     dag_id=f"{HYPERCOMPUTER}_recipes_{MODEL_ID}_{FRAMEWORK}",
     schedule=SCHEDULED_TIME,
     tags=[
-        "simple",
-        "aotc",
-        "nightly",
         "reproducibility",
         "experimental",
         "xlml",
+        "regressiontests",
+        "a3mega",
     ],
     start_date=datetime.datetime(2024, 11, 15),
     catchup=False,

diff --git a/dags/map_reproducibility/a3mega_llama_3_1_70b_nemo.py b/dags/map_reproducibility/a3mega_llama_3_1_70b_nemo.py
@@ -45,7 +45,7 @@
 from dags.map_reproducibility.utils.common_utils import get_recipe_repo_path
 
 
-# Run once a day at 2 pm UTC (6 am PST)
+# Run once a day at 11 am UTC (3 am PST)
 SCHEDULED_TIME = "0 11 * * *" if composer_env.is_prod_env() else None
 
 MODEL_ID = "llama-3.1-70b"
@@ -167,12 +167,11 @@ def run_aotc_workload():
     dag_id=f"{HYPERCOMPUTER}_recipes_{MODEL_ID}_{FRAMEWORK}",
     schedule=SCHEDULED_TIME,
     tags=[
-        "simple",
-        "aotc",
-        "nightly",
         "reproducibility",
         "experimental",
         "xlml",
+        "regressiontests",
+        "a3mega",
     ],
     start_date=datetime.datetime(2024, 11, 15),
     catchup=False,

diff --git a/dags/map_reproducibility/a3mega_llama_3_70b_nemo.py b/dags/map_reproducibility/a3mega_llama_3_70b_nemo.py
@@ -45,7 +45,7 @@
 from dags.map_reproducibility.utils.common_utils import get_recipe_repo_path
 
 
-# Run once a day at 2 pm UTC (6 am PST)
+# Run once a day at 12 pm UTC (4 am PST)
 SCHEDULED_TIME = "0 12 * * *" if composer_env.is_prod_env() else None
 
 MODEL_ID = "llama-3-70b"
@@ -166,12 +166,11 @@ def run_aotc_workload():
     dag_id=f"{HYPERCOMPUTER}_recipes_{MODEL_ID}_{FRAMEWORK}",
     schedule=SCHEDULED_TIME,
     tags=[
-        "simple",
-        "aotc",
-        "nightly",
         "reproducibility",
         "experimental",
         "xlml",
+        "regressiontests",
+        "a3mega",
     ],
     start_date=datetime.datetime(2024, 11, 15),
     catchup=False,

diff --git a/dags/map_reproducibility/a3mega_mixtral_8_7b_nemo.py b/dags/map_reproducibility/a3mega_mixtral_8_7b_nemo.py
@@ -45,7 +45,7 @@
 from dags.map_reproducibility.utils.common_utils import get_recipe_repo_path
 
 
-# Run once a day at 2 pm UTC (6 am PST)
+# Run once a day at 10 am UTC (2 am PST)
 SCHEDULED_TIME = "0 10 * * *" if composer_env.is_prod_env() else None
 
 MODEL_ID = "mixtral-8x7b"
@@ -167,12 +167,11 @@ def run_aotc_workload():
     dag_id=f"{HYPERCOMPUTER}_recipes_{MODEL_ID}_{FRAMEWORK}",
     schedule=SCHEDULED_TIME,
     tags=[
-        "simple",
-        "aotc",
-        "nightly",
         "reproducibility",
         "experimental",
         "xlml",
+        "regressiontests",
+        "a3mega",
     ],
     start_date=datetime.datetime(2024, 11, 15),
     catchup=False,

diff --git a/dags/map_reproducibility/a3ultra_llama_3_1_70b_nemo.py b/dags/map_reproducibility/a3ultra_llama_3_1_70b_nemo.py
@@ -170,12 +170,12 @@ def run_aotc_workload():
     dag_id=f"{HYPERCOMPUTER}_recipes_{MODEL_ID}_{FRAMEWORK}",
     schedule=SCHEDULED_TIME,
     tags=[
-        "simple",
-        "aotc",
-        "nightly",
         "reproducibility",
         "experimental",
         "xlml",
+        "regressiontests",
+        "a3mega",
+        "aotc",
     ],
     start_date=datetime.datetime(2024, 11, 15),
     catchup=False,

diff --git a/dags/map_reproducibility/a3ultra_mixtral_8_7b_nemo.py b/dags/map_reproducibility/a3ultra_mixtral_8_7b_nemo.py
@@ -44,8 +44,8 @@
 from dags.map_reproducibility.utils.common_utils import get_bq_writer_path
 from dags.map_reproducibility.utils.common_utils import get_recipe_repo_path
 
-# Run once a day at 2 pm UTC (6 am PST)
-SCHEDULED_TIME = "0 14 * * *" if composer_env.is_prod_env() else None
+# Run once a day at 12 pm UTC (4 am PST)
+SCHEDULED_TIME = "0 12 * * *" if composer_env.is_prod_env() else None
 
 MODEL_ID = "mixtral-8x7b"
 METRICS_MODEL_ID = "mixtral-7b"
@@ -166,12 +166,11 @@ def run_aotc_workload():
     dag_id=f"{HYPERCOMPUTER}_recipes_{MODEL_ID}_{FRAMEWORK}",
     schedule=SCHEDULED_TIME,
     tags=[
-        "simple",
-        "aotc",
-        "nightly",
-        "reproducibility",
+       "reproducibility",
         "experimental",
         "xlml",
+        "regressiontests",
+        "a3ultra",
     ],
     start_date=datetime.datetime(2024, 11, 15),
     catchup=False,