dags/map_reproducibility/internal_runs/sample_a3mega_maxtext_single_run.py [19:65]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
script_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.abspath(os.path.join(script_dir, "..", "..", ".."))

print(f"Script directory: {script_dir}")
print(f"Project root: {project_root}")

if project_root not in sys.path:
  sys.path.insert(0, project_root)

import datetime
from dags.map_reproducibility.utils.constants import Image
from dags.map_reproducibility.internal_runs.dag_configs import DAG_CONFIGS_MEGA
from dags.map_reproducibility.utils.sample_workload_utils import run_internal_sample_aotc_workload


# Skip execution when being run as part of the DAG check
# Checking if the file doesn't exist is a reliable way to detect this context
base_recipe_repo_root = os.path.abspath(
    os.path.join(
        os.path.dirname(os.path.abspath(__file__)),
        "..",
        "..",
        "..",
        "..",
        "internal-gpu-recipes",
    )
)

if not os.path.exists(base_recipe_repo_root):
  print(
      f"Skipping sample_a3ultra_maxtext_single_run.py - required directory not found: {base_recipe_repo_root}"
  )
  sys.exit(0)


def main():
  utc_date = datetime.datetime.now(datetime.timezone.utc).strftime("%Y-%m-%d")
  NIGHTLY_IMAGE = f"{Image.MAXTEXT_JAX_STABLE_NIGHTLY}:{utc_date}"
  RELEASE_IMAGE = f"{Image.MAXTEXT_JAX_STABLE_RELEASE}:{utc_date}"

  # The dynamic date-based image is defined above, but can be overridden below
  # with specific image versions and bucket names as required for testing
  RELEASE_IMAGE = f"{Image.MAXTEXT_JAX_STABLE_RELEASE}:2025-04-17"
  SAMPLE_RUN_BUCKET_NAME = "yujunzou-dev-supercomputer-testing"

  # Setup configuration
  relative_config_yaml_path = (
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



dags/map_reproducibility/internal_runs/sample_a3ultra_maxtext_single_run.py [19:64]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
script_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.abspath(os.path.join(script_dir, "..", "..", ".."))

print(f"Script directory: {script_dir}")
print(f"Project root: {project_root}")

if project_root not in sys.path:
  sys.path.insert(0, project_root)

import datetime
from dags.map_reproducibility.utils.constants import Image
from dags.map_reproducibility.internal_runs.dag_configs import DAG_CONFIGS_ULTRA
from dags.map_reproducibility.utils.sample_workload_utils import run_internal_sample_aotc_workload

# Skip execution when being run as part of the DAG check
# Checking if the file doesn't exist is a reliable way to detect this context
base_recipe_repo_root = os.path.abspath(
    os.path.join(
        os.path.dirname(os.path.abspath(__file__)),
        "..",
        "..",
        "..",
        "..",
        "internal-gpu-recipes",
    )
)

if not os.path.exists(base_recipe_repo_root):
  print(
      f"Skipping sample_a3ultra_maxtext_single_run.py - required directory not found: {base_recipe_repo_root}"
  )
  sys.exit(0)


def main():
  utc_date = datetime.datetime.now(datetime.timezone.utc).strftime("%Y-%m-%d")
  NIGHTLY_IMAGE = f"{Image.MAXTEXT_JAX_STABLE_NIGHTLY}:{utc_date}"
  RELEASE_IMAGE = f"{Image.MAXTEXT_JAX_STABLE_RELEASE}:{utc_date}"

  # The dynamic date-based image is defined above, but can be overridden below
  # with specific image versions and bucket names as required for testing
  RELEASE_IMAGE = f"{Image.MAXTEXT_JAX_STABLE_RELEASE}:2025-04-17"
  SAMPLE_RUN_BUCKET_NAME = "yujunzou-dev-supercomputer-testing"

  # Setup configuration
  relative_config_yaml_path = (
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -