Skip to content

Commit ca85778

Browse files
Ronald Tseronaldtse
authored andcommitted
fix: torch importorskip in the stitch test (CI has no torch)
1 parent 8118371 commit ca85778

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

‎tests/test_stitch.py‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,11 @@
22
config, preserving the top singular subspace (the microkimi
33
closed-form bridge)."""
44

5-
import torch
5+
import pytest
66

7-
from src.gpu.modal_distill import svd_stitch_state
7+
pytest.importorskip("torch")
8+
9+
from src.gpu.modal_distill import svd_stitch_state # noqa: E402
810

911

1012
def _wide_narrow():

0 commit comments

Comments
 (0)