Bugfix with missing num_steps parameters in CorrDiff generate.py

[CharlelieLrt]

PhysicsNeMo Pull Request

Description

Checklist

• I am familiar with the Contributing Guidelines.
• New or existing tests cover these changes.
• The documentation is up to date with these changes.
• The CHANGELOG.md is up to date with these changes.
• An issue is linked to this pull request.
• If I am implementing a new model or modifying any existing model, I have followed the Models Implementation Coding Standards.

Dependencies

Review Process

All PRs are reviewed by the PhysicsNeMo team before merging.

Depending on which files are changed, GitHub may automatically assign a maintainer for review.

We are also testing AI-based code review tools (e.g., Greptile), which may add automated comments with a confidence score.
This score reflects the AI’s assessment of merge readiness and is not a qualitative judgment of your work, nor is
it an indication that the PR will be accepted / rejected.

AI-generated feedback should be reviewed critically for usefulness.
You are not required to respond to every AI comment, but they are intended to help both authors and reviewers.
Please react to Greptile comments with 👍 or 👎 to provide feedback on their accuracy.

[CharlelieLrt]

/blossom-ci

[greptile-apps]

Greptile Summary

Adds missing num_steps parameter to stochastic sampler config and defensive defaults in code.

Changes:

• Added num_steps: 18 to stochastic.yaml (matching the default in code)
• Updated generate.py to use getattr() with fallback defaults (9 for deterministic, 18 for stochastic) to prevent crashes when num_steps is missing from config
• Reorganized imports alphabetically in generate.py

Issues:

• YAML comment indentation is incorrect (indented instead of at column 0)

Important Files Changed

Filename	Overview
examples/weather/corrdiff/conf/base/generation/sampler/stochastic.yaml	Added missing num_steps: 18 parameter with comment, but comment has incorrect indentation
examples/weather/corrdiff/generate.py	Added getattr() with defaults to handle missing num_steps gracefully, imports reorganized alphabetically

_{Last reviewed commit: 3b43852}

[greptile-apps]

_{2 files reviewed, 1 comment}

_{Edit Code Review Agent Settings | Greptile}

[coreyjadams]

LGTM!

[CharlelieLrt]

/blossom-ci

[CharlelieLrt]

/blossom-ci