Auto activate venv when python-version is set

astral-sh · Dec 13, 2024 · a3f221d · a3f221d
1 parent 856099c
commit a3f221d
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 9 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -155,5 +155,4 @@ jobs:
           if [ "$UV_PYTHON" != "${{ matrix.python-version }}" ]; then
             exit 1
           fi
-      - run: uv sync
-        working-directory: __tests__/fixtures/uv-project
+      - run: uv pip install --python=${{ matrix.python-version }} -i https://pypi.anaconda.org/scientific-python-nightly-wheels/simple cython
diff --git a/dist/setup/index.js b/dist/setup/index.js
diff --git a/src/setup-uv.ts b/src/setup-uv.ts
@@ -23,6 +23,7 @@ import {
   toolDir,
   version,
 } from "./utils/inputs";
+import * as exec from "@actions/exec";
 
 async function run(): Promise<void> {
   const platform = getPlatform();
@@ -46,7 +47,7 @@ async function run(): Promise<void> {
     addUvToPath(setupResult.uvDir);
     addToolBinToPath();
     setToolDir();
-    setupPython();
+    await setupPython();
     addMatchers();
     setCacheDir(cacheLocalPath);
 
@@ -125,10 +126,17 @@ function setToolDir(): void {
   }
 }
 
-function setupPython(): void {
+async function setupPython(): Promise<void> {
   if (pythonVersion !== "") {
     core.exportVariable("UV_PYTHON", pythonVersion);
     core.info(`Set UV_PYTHON to ${pythonVersion}`);
+    const options: exec.ExecOptions = {
+      silent: !core.isDebug(),
+    };
+    const execArgs = ["venv", "--python", pythonVersion];
+
+    core.info("Activating python venv...");
+    await exec.exec("uv", execArgs, options);
   }
 }