fix: import torch after the skip guard

Author: Ronald Tse

tests/test_stitch.py

@@ -5,6 +5,7 @@
 import pytest
 
 pytest.importorskip("torch")
+import torch  # noqa: E402
 
 from src.gpu.modal_distill import svd_stitch_state  # noqa: E402