Add resolver to persist meta ttl key

polyaxon · Oct 3, 2024 · 9d033fb · 9d033fb
1 parent 86285b2
commit 9d033fb
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/haupt/haupt/orchestration/scheduler/resolver.py b/haupt/haupt/orchestration/scheduler/resolver.py
@@ -46,6 +46,7 @@
     META_IS_EXTERNAL,
     META_PORTS,
     META_REWRITE_PATH,
+    META_TTL,
     META_UPLOAD_ARTIFACTS,
 )
 from polyaxon._contexts import keys as ctx_keys
@@ -676,6 +677,12 @@ def _persist_meta_info(self):
         # handle services ports
         if self.compiled_operation.is_service_run and self.compiled_operation.run.ports:
             self.run.meta_info[META_PORTS] = self.compiled_operation.run.ports
+        # handle ttl
+        if (
+            self.compiled_operation.termination
+            and self.compiled_operation.termination.ttl
+        ):
+            self.run.meta_info[META_TTL] = self.compiled_operation.termination.ttl
 
     def _persist_resources(self):
         if self.compiled_operation.has_pipeline: