scripts/show_features.py

"""Print the supported and unavailable project features."""


COMPLETED_FEATURES = [
    "Compact global-grid forecaster construction and checkpoint loading",
    "Single-GPU and single-node DDP training",
    "Offline local checkpoint inference",
    "Synthetic smoke-data generation",
    "FV3GFS NetCDF schema validation",
    "Separate data, weight, result, and work directories",
    "Configurable model name, data paths, checkpoint paths, and output paths",
    "AMD DCU-compatible model loading and smoke validation",
]

PENDING_FEATURES = [
    "Full SFNO and DYffusion architecture integration",
    "Official Hugging Face checkpoint compatibility",
    "Full FV3GFS validation rollout, pending requester-pays data",
    "Paper-scale 100-year training and 10-year evaluation",
    "Paper-matched 25-member climate metrics",
]


def main() -> None:
    print("Spherical-DYffusion-SFNO project features")
    print("Completed:")
    for index, feature in enumerate(COMPLETED_FEATURES, start=1):
        print(f"  {index}. {feature}")
    print("Pending external inputs or resources:")
    for index, feature in enumerate(PENDING_FEATURES, start=1):
        print(f"  {index}. {feature}")


if __name__ == "__main__":
    main()