chore: create alias to source sky env in dev mode

skypilot-org · Jan 14, 2025 · 96f269c · 96f269c
1 parent 35f0cf4
commit 96f269c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/sky/templates/jobs-controller.yaml.j2 b/sky/templates/jobs-controller.yaml.j2
@@ -26,6 +26,10 @@ setup: |
   echo 'export SKYPILOT_DEV=1' >> ~/.bashrc
   {% endif %}
 
+  {% if controller_envs.get("SKYPILOT_DEV") %}
+  grep -q "alias sky-env=" ~/.bashrc || echo 'alias sky-env="{{ sky_activate_python_env }}"' >> ~/.bashrc
+  {% endif %}
+
   # Create systemd service file
   mkdir -p ~/.config/systemd/user/
 

diff --git a/sky/templates/sky-serve-controller.yaml.j2 b/sky/templates/sky-serve-controller.yaml.j2
@@ -15,6 +15,10 @@ setup: |
   {{cmd}}
   {%- endfor %}
 
+  {% if controller_envs.get("SKYPILOT_DEV") %}
+  grep -q "alias sky-env=" ~/.bashrc || echo 'alias sky-env="{{ sky_activate_python_env }}"' >> ~/.bashrc
+  {% endif %}
+
   # Install serve dependencies.
   # TODO(tian): Gather those into serve constants.
   pip list | grep uvicorn > /dev/null 2>&1 || pip install uvicorn > /dev/null 2>&1