diff --git a/src/anemoi/inference/checkpoint/metadata/__init__.py b/src/anemoi/inference/checkpoint/metadata/__init__.py index f961a98..433ab35 100644 --- a/src/anemoi/inference/checkpoint/metadata/__init__.py +++ b/src/anemoi/inference/checkpoint/metadata/__init__.py @@ -386,10 +386,10 @@ def summary(self): print(sorted(self.accumulations_params)) print() - print("Select:") - print(json.dumps(self.select, indent=2)) - print() + # print("Select:") + # print(json.dumps(self.select, indent=2)) + # print() - print("Order by:") - print(json.dumps(self.order_by, indent=2)) - print() + # print("Order by:") + # print(json.dumps(self.order_by, indent=2)) + # print() diff --git a/src/anemoi/inference/plugin.py b/src/anemoi/inference/plugin.py index 086e458..52a2bdc 100644 --- a/src/anemoi/inference/plugin.py +++ b/src/anemoi/inference/plugin.py @@ -80,15 +80,15 @@ def _output(self, *args, **kwargs): @property def param_sfc(self): - return self.runner.checkpoint.param_sfc + return self.runner.param_sfc @property def param_level_pl(self): - return self.runner.checkpoint.param_level_pl + return self.runner.param_level_pl @property def param_level_ml(self): - return self.runner.checkpoint.param_level_ml + return self.runner.param_level_ml @property def constant_fields(self): diff --git a/src/anemoi/inference/runner.py b/src/anemoi/inference/runner.py index 1b60eae..1890592 100644 --- a/src/anemoi/inference/runner.py +++ b/src/anemoi/inference/runner.py @@ -419,6 +419,30 @@ def lagged(self): result = [-s * self.hour_steps for s in result] return sorted(result) + @property + def param_sfc(self): + param_sfc = self.checkpoint.param_sfc + + # Remove diagnostic params + + param_sfc = [p for p in param_sfc if p not in self.checkpoint.diagnostic_params] + + return param_sfc + + @property + def param_level_pl(self): + + # To do remove diagnostic params + + return self.checkpoint.param_level_pl + + @property + def param_level_ml(self): + + # To do remove diagnostic params + + return self.checkpoint.param_level_ml + class DefaultRunner(Runner): """_summary_