fix: Add missing session creation args (starts_at, bootstrap_script) (#…

…213)
lablup · Mar 20, 2022 · 99cae84 · 99cae84
1 parent 07f59d4
commit 99cae84
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/changes/213.fix.md b/changes/213.fix.md
@@ -0,0 +1 @@
+Fix missing propagation of `starts_at` and `bootstrap_script` arguments to the session creation API, which have been supported since v20.03
diff --git a/src/ai/backend/client/func/session.py b/src/ai/backend/client/func/session.py
@@ -278,6 +278,8 @@ async def get_or_create(
         else:
             params['config']['clusterSize'] = cluster_size
         if api_session.get().api_version >= (5, '20191215'):
+            params['starts_at'] = starts_at
+            params['bootstrap_script'] = bootstrap_script
             if assign_agent is not None:
                 params['config'].update({
                     'mount_map': mount_map,