feat: Execute keypair bootstrap script for batch compute session (#437)

lablup · Jun 6, 2022 · dcdf9c4 · dcdf9c4
1 parent ed6cd9e
commit dcdf9c4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/changes/437.feature b/changes/437.feature
@@ -0,0 +1 @@
+Execute the keypair bootstrap script for batch compute session as well (previously it was only executed for interactive sessions)
diff --git a/src/ai/backend/manager/api/session.py b/src/ai/backend/manager/api/session.py
@@ -506,8 +506,7 @@ async def _create(request: web.Request, params: dict[str, Any]) -> web.Response:
         owner_uuid, group_id, resource_policy = await _query_userinfo(request, params, conn)
 
         # Use keypair bootstrap_script if it is not delivered as a parameter
-        # (only for INTERACTIVE sessions).
-        if params['session_type'] == SessionTypes.INTERACTIVE and not params['bootstrap_script']:
+        if not params['bootstrap_script']:
             script, _ = await query_bootstrap_script(conn, owner_access_key)
             params['bootstrap_script'] = script