From 02cdb35a6929e38aa9be8edc8d4c8a576229d6bb Mon Sep 17 00:00:00 2001 From: Ronald Tse Date: Wed, 2 Sep 2026 17:51:51 +0200 Subject: [PATCH] fix: importorskip torch in the MTP test (CI installs neither torch nor modal) --- tests/test_gpu_mtp.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/test_gpu_mtp.py b/tests/test_gpu_mtp.py index acb15e2..ab03a61 100644 --- a/tests/test_gpu_mtp.py +++ b/tests/test_gpu_mtp.py @@ -1,3 +1,7 @@ +import pytest + +pytest.importorskip("torch") + import torch from gpu.mtp import MTPHead, mtp_named