Skip to content

Commit 02cdb35

Browse files
author
Ronald Tse
committed
fix: importorskip torch in the MTP test (CI installs neither torch nor modal)
1 parent 0d7ee64 commit 02cdb35

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

‎tests/test_gpu_mtp.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,7 @@
1+
import pytest
2+
3+
pytest.importorskip("torch")
4+
15
import torch
26

37
from gpu.mtp import MTPHead, mtp_named

0 commit comments

Comments
 (0)