diff --git a/docs/platforms/aurora_operations.md b/docs/platforms/aurora_operations.md index d0f1726..e969b3f 100644 --- a/docs/platforms/aurora_operations.md +++ b/docs/platforms/aurora_operations.md @@ -45,7 +45,7 @@ cat $PBS_NODEFILE Open a new terminal on the UAN and run: ```bash -cd /lus/flare/projects/ModCon/ngetty/PRISM +cd /lus/flare/projects///PRISM # Quick test (5 steps, 1B model): python tools/launch_aurora_web.py \ diff --git a/docs/platforms/running_on_polaris.md b/docs/platforms/running_on_polaris.md index 8e1797f..189c364 100644 --- a/docs/platforms/running_on_polaris.md +++ b/docs/platforms/running_on_polaris.md @@ -98,7 +98,7 @@ recommended layout: datasets//shards/*.tar models// # HF snapshot dirs envs/.venv-polaris/ # built by setup_polaris_env.sh - repos/PRISM/ # checkout + repos/PRISM/ # checkout logs// # PBS stdout/stderr ``` diff --git a/tests/test_scits_fix_static.py b/tests/test_scits_fix_static.py index 56867ab..8bbd480 100644 --- a/tests/test_scits_fix_static.py +++ b/tests/test_scits_fix_static.py @@ -455,7 +455,7 @@ def test_no_user_specific_hub_paths_in_model_configs(): Tokenizer paths under a project dir are still allowed: the interleaved TS configs legitimately point at custom tokenizers checked in under - /flare/ModCon//BaseMM_PRISM/tokenizers/. + /flare///PRISM/tokenizers/. """ offenders = [] for cfg in sorted((REPO_ROOT / "src" / "conf" / "model").glob("*.yaml")): diff --git a/tools/ci/site_path_allowlist.json b/tools/ci/site_path_allowlist.json index 35ee1f2..321c339 100644 --- a/tools/ci/site_path_allowlist.json +++ b/tools/ci/site_path_allowlist.json @@ -20,7 +20,7 @@ "docs/modalities/image_decoder/docci_data.md": 3, "docs/modalities/timeseries.md": 6, "docs/models/auroragpt_vlm.md": 1, - "docs/platforms/aurora_operations.md": 4, + "docs/platforms/aurora_operations.md": 3, "docs/platforms/daos_setup.md": 10, "docs/platforms/running_on_polaris.md": 8, "docs/results/patrick_tsqa_ab.md": 3, @@ -50,7 +50,7 @@ "tools/isoflop_collect.py": 1, "tools/isoflop_launch.py": 1, "tools/isoflop_plan.py": 4, - "tools/launch_aurora.py": 3, + "tools/launch_aurora.py": 2, "tools/launch_aurora_daos.py": 2, "tools/launch_aurora_web.py": 1, "tools/launch_baremetal.py": 2, diff --git a/tools/launch_aurora.py b/tools/launch_aurora.py index d1df0d3..d59c87b 100644 --- a/tools/launch_aurora.py +++ b/tools/launch_aurora.py @@ -56,7 +56,10 @@ def main(): env_config = load_dotenv() # Defaults from .env or hardcoded fallbacks - default_prism_dir = env_config.get("PRISM_DIR", "/flare/ModCon/ngetty/BaseMM_PRISM") + # Fall back to the checkout this launcher lives in, not a hardcoded user path. + default_prism_dir = env_config.get( + "PRISM_DIR", os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + ) default_hf_home = env_config.get("HF_HOME", "/flare/ModCon/sandeep") env_config.get("PROJECT_ALLOCATION", "ModCon") env_config.get("QUEUE_NAME", "debug") diff --git a/tools/launch_polaris.py b/tools/launch_polaris.py index 3f39005..676762b 100755 --- a/tools/launch_polaris.py +++ b/tools/launch_polaris.py @@ -239,7 +239,7 @@ def main() -> int: env_config = load_dotenv() # When --prism-dir isn't given, prefer the launcher's own location over - # cwd. That way `ssh polaris "cd /eagle/.../BaseMM_PRISM && \ + # cwd. That way `ssh polaris "cd /eagle/.../PRISM && \ # python tools/launch_polaris.py ..."` Just Works from a different host. if args.prism_dir: prism_dir = args.prism_dir