allow to set restart_policy when creating onprem

karmab · Jun 5, 2024 · 3bc95aa · 3bc95aa
1 parent e9d8db1
commit 3bc95aa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ailib/common/__init__.py b/ailib/common/__init__.py
@@ -167,6 +167,9 @@ def create_onprem(overrides={}, debug=False):
                 response = urllib.request.urlopen(pod_url).read().decode('utf-8')
                 response = response.replace('latest', onprem_version).replace(f'centos7:{onprem_version}',
                                                                               'centos7:latest')
+                restart_policy = overrides.get('restart_policy')
+                if restart_policy is not None and restart_policy == 'Always':
+                    response.replace('Never', 'Always')
                 p.write(response)
         if os.path.exists('configmap.yml'):
             info("Using existing configmap.yml")