refactor(github): update DAG and utils to use trigger_github_repository_dispatch as primary name

Author: RexBearIU

dags/multipod/maxtext_e2e_tests.py

@@ -23,7 +23,7 @@
 from airflow.models.param import Param
 from airflow.operators.trigger_dagrun import TriggerDagRunOperator
 from airflow.utils.trigger_rule import TriggerRule
-from xlml.utils.github import fire_github_callback, validate_git_trigger
+from xlml.utils.github import trigger_github_repository_dispatch, validate_git_trigger
 
 with models.DAG(
     dag_id="maxtext_e2e_tests",
@@ -79,12 +79,12 @@
       poke_interval=600,  # check every 10 minutes for child DAG completion
   )
 
-  github_callback_pre_training = fire_github_callback.override(
+  github_callback_pre_training = trigger_github_repository_dispatch.override(
       task_id="fire_github_callback_pre_training",
       trigger_rule=TriggerRule.ALL_SUCCESS,
   )(test_type="pre_training")
 
-  github_callback_post_training = fire_github_callback.override(
+  github_callback_post_training = trigger_github_repository_dispatch.override(
       task_id="fire_github_callback_post_training",
       trigger_rule=TriggerRule.ALL_SUCCESS,
   )(test_type="post_training")

xlml/utils/github.py

@@ -44,7 +44,7 @@ def validate_git_trigger() -> None:
 
 
 @task
-def fire_github_callback(
+def trigger_github_repository_dispatch(
     test_type: str | None = None,
     repo: str | None = None,
     token: str | None = None,
@@ -99,4 +99,4 @@ def fire_github_callback(
   response.raise_for_status()
 
 
-trigger_github_repository_dispatch = fire_github_callback
+fire_github_callback = trigger_github_repository_dispatch