Adding correct path for module not found error while collecting test (#19778)

Co-authored-by: sys-lpot-val <sys_lpot_val@intel.com>
2026-03-07 05:56:16 +05:30
parent e818f8219a
commit a0d085c16d
6 changed files with 8 additions and 13 deletions
--- a/python/sglang/multimodal_gen/csrc/attn/vmoba_attn/tests/test_vmoba_attn.py
+++ b/python/sglang/multimodal_gen/csrc/attn/vmoba_attn/tests/test_vmoba_attn.py
@@ -4,8 +4,7 @@ import random

 import pytest
 import torch
-from csrc.attn.vmoba_attn.vmoba import moba_attn_varlen
-
+from sglang.multimodal_gen.csrc.attn.vmoba_attn.vmoba import moba_attn_varlen

 def generate_test_data(
    batch_size, total_seqlen, num_heads, head_dim, dtype, device="cuda"