Add output finalise

ecmwf-lab · Jun 21, 2024 · 7a07b73 · 7a07b73
1 parent 5777942
commit 7a07b73
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ai_models/model.py b/src/ai_models/model.py
@@ -131,6 +131,8 @@ def collect_archive_requests(self, written):
             self.archiving[path].add(handle.as_mars())
 
     def finalise(self):
+        self.output.finalise()
+
         if self.archive_requests:
             with open(self.archive_requests, "w") as f:
                 json_requests = []