Skip to content

Commit 754aadc

Browse files
committed
Use base create_output
1 parent 7bd9d9b commit 754aadc

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/anemoi/inference/runners/parallel.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -153,17 +153,15 @@ def predict_step(self, model: Any, input_tensor_torch: torch.Tensor, **kwargs: A
153153
raise err
154154

155155
def create_output(self) -> Output:
156-
"""Creates the output.
156+
"""Creates the real output on rank 0 and a `none` on the others.
157157
158158
Returns
159159
-------
160160
Output
161161
The created output.
162162
"""
163163
if self.global_rank == 0:
164-
output = create_output(self, self.config.output)
165-
LOG.info("Output: %s", output)
166-
return output
164+
return super().create_output()
167165
else:
168166
output = create_output(self, "none")
169167
return output

0 commit comments

Comments
 (0)