From 0bb03e069dc9c3ccbfb10d812575477f397b0480 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 15 May 2026 05:27:46 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/training/worker.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py index 279d42c095..2237d07a84 100644 --- a/studio/backend/core/training/worker.py +++ b/studio/backend/core/training/worker.py @@ -385,6 +385,7 @@ def _ensure_tilelang_backend(event_queue: Any, model_name: str) -> None: try: import tilelang # noqa: F401 import tvm_ffi # noqa: F401 + logger.info("tilelang + apache-tvm-ffi already installed") return except ImportError: @@ -407,13 +408,19 @@ def _ensure_tilelang_backend(event_queue: Any, model_name: str) -> None: ] if shutil.which("uv"): pypi_cmd = [ - "uv", "pip", "install", - "--python", sys.executable, + "uv", + "pip", + "install", + "--python", + sys.executable, *specs, ] else: pypi_cmd = [ - sys.executable, "-m", "pip", "install", + sys.executable, + "-m", + "pip", + "install", *specs, ]