Skip to content

Commit

Permalink
Merge branch 'develop' of github.com:ecmwf/anemoi-inference into develop
Browse files Browse the repository at this point in the history
  • Loading branch information
b8raoult committed May 25, 2024
2 parents b791c8b + 2bab331 commit 41b71aa
Showing 1 changed file with 22 additions and 1 deletion.
23 changes: 22 additions & 1 deletion src/anemoi/inference/commands/checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,28 @@ def add_arguments(self, command_parser):
command_parser.add_argument("path", help="Path to the checkpoint.")

def run(self, args):
Checkpoint(args.path).describe()
from anemoi.utils.text import dotted_line

c = Checkpoint(args.path)
print("num_input_features:", c.num_input_features)
print("hour_steps:", c.hour_steps)
result = list(range(0, c.multi_step))
result = [-s * c.hour_steps for s in result]
print(sorted(result))
print("multi_step:", c.multi_step)
print()
print("MARS requests:")
print(dotted_line())
print("param_sfc:", c.param_sfc)
print("param_level_pl:", c.param_level_pl)
print("param_level_ml:", c.param_level_ml)
print("prognostic_params:", c.prognostic_params)
print("diagnostic_params:", c.diagnostic_params)
print("constants_from_input:", c.constants_from_input)
print("computed_constants:", c.computed_constants)
print("computed_forcings:", c.computed_forcings)
print()
print("imputable variables", c.imputable_variables)


command = CheckpointCmd

0 comments on commit 41b71aa

Please sign in to comment.