diff --git a/model/registry.yaml b/model/registry.yaml index e240b35638..1f5b6d7e7a 100644 --- a/model/registry.yaml +++ b/model/registry.yaml @@ -603,7 +603,7 @@ models: - Generated by scripts/training/train_predictive_planner.py - model_id: predictive_proxy_selected_v1 display_name: Predictive planner model (predictive_proxy_selected_v1) - local_path: output/tmp/predictive_planner/training/predictive_proxy_selected_v1/predictive_model.pt + local_path: output/model_cache/predictive_proxy_selected_v1/predictive_proxy_selected_v1-predictive_model.pt config_path: null commit: dfc4aea84e25cc83f9888c620286457bab3e1596 wandb_run_id: geedo1po @@ -650,7 +650,7 @@ models: - Generated by scripts/training/train_predictive_planner.py - model_id: predictive_proxy_selected_v2_full display_name: Predictive planner model (predictive_proxy_selected_v2_full) - local_path: output/tmp/predictive_planner/pipeline/predictive_br07_all_maps_randomized_full_20260305T123116Z/training/predictive_model.pt + local_path: output/model_cache/predictive_proxy_selected_v2_full/predictive_proxy_selected_v2_full-predictive_model.pt config_path: configs/training/predictive/predictive_br07_all_maps_randomized_full.yaml commit: cef93136b92ddca9b0c4436bc44049412461a2fd wandb_run_id: u40parjb diff --git a/tests/unit/test_registry.py b/tests/unit/test_registry.py index 9a2686bd81..d522734213 100644 --- a/tests/unit/test_registry.py +++ b/tests/unit/test_registry.py @@ -1000,3 +1000,15 @@ def test_upsert_registry_entry_requires_model_id(tmp_path: Path) -> None: {"local_path": "output/model_cache/demo/model.zip"}, registry_path=tmp_path / "registry.yaml", ) + + +def test_predictive_proxy_registry_paths_match_release_assets() -> None: + """Predictive proxy cache pointers should name their release assets exactly.""" + registry_path = Path(__file__).resolve().parents[2] / "model" / "registry.yaml" + entries = registry.load_registry(registry_path) + + for model_id in ("predictive_proxy_selected_v1", "predictive_proxy_selected_v2_full"): + entry = entries[model_id] + release = entry["github_release"] + assert Path(entry["local_path"]).parts[:3] == ("output", "model_cache", model_id) + assert Path(entry["local_path"]).name == release["asset_name"]