File size: 2,961 Bytes
939693f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | #!/usr/bin/env python3
"""Check that the FireWx-FM release artifacts are present."""
from __future__ import annotations
from pathlib import Path
REQUIRED = [
"firewxfm/modeling_unet.py",
"firewxfm/tiled_inference.py",
"firewxfm/serve_conus.py",
"models/metadata/input_channels.json",
"models/metadata/final_model_manifest.json",
"models/metadata/input_normalization_stats.json",
"models/checkpoints/firewxfm_conus_lowposw_noexposure_seed42.pt",
"models/checkpoints/firewxfm_2024_staticfix_region_balanced_bce_seed42.pt",
"examples/final_prediction/firewxfm_conus_20260706_t12_usgs_calibrated_probability_5km_lower48.tif",
"examples/final_prediction/firewxfm_conus_20260706_t12_usgs_calibrated_heatmap.png",
"examples/final_prediction/firewxfm_conus_20260706_t12_usgs_calibrated_heatmap_rgb.tif",
"examples/final_prediction/firewxfm_conus_20260706_t12_usgs_calibrated_summary.json",
]
OPTIONAL = [
"models/metadata/run_summary.json",
"models/metadata/tile_summary.json",
"docs/split_design.md",
"docs/reproducibility_manifest.md",
"docs/inference_contract.md",
"docs/serving_notes.md",
"docs/model_card.md",
"REPRODUCIBILITY.md",
"paper/Does_Your_Wildfire_Prediction_Model_ICLR_revision.pdf",
"paper/iclr_source/main.tex",
"technical_report/FireWx-FM_technical_report.pdf",
"technical_report/source/main.tex",
"scripts/audit_firewxfm_phase0.py",
"scripts/prepare_firms_target_hour_csvs.py",
"scripts/smoke_inference.py",
"training/configs/stage1_cache_conus_multiyear_template.json",
"training/configs/train_firewxfm_conus_multiyear_template.json",
"training/configs/train_firewxfm_multitask_pretrain_template.json",
"training/configs/eval_region_disjoint_template.json",
"training/configs/baseline_suite_template.json",
"scripts/build_5km_input_stack.py",
"scripts/audit_firewxfm_longitude_seams.py",
"scripts/make_firewxfm_current_static_corrected_stack.py",
"scripts/apply_prediction_mask.py",
"scripts/write_prediction_tif.py",
]
def describe(path: Path) -> str:
if not path.exists():
return "missing"
if path.is_file():
size_mb = path.stat().st_size / (1024 * 1024)
return f"ok ({size_mb:.2f} MB)"
return "ok"
def main() -> int:
root = Path(__file__).resolve().parents[1]
missing = []
print("Required artifacts")
for rel in REQUIRED:
status = describe(root / rel)
print(f" {rel}: {status}")
if status == "missing":
missing.append(rel)
print("\nOptional artifacts")
for rel in OPTIONAL:
print(f" {rel}: {describe(root / rel)}")
if missing:
print("\nRelease check failed. Missing required files:")
for rel in missing:
print(f" - {rel}")
return 1
print("\nRelease check passed.")
return 0
if __name__ == "__main__":
raise SystemExit(main())
|