Skip to content

Commit 398dc26

Browse files
authored
bugfix: Change module path in test_pod_kernels.py (#1842)
<!-- .github/pull_request_template.md --> ## 📌 Description The series of code refactgor for tvm-ffi (e.g. [PR 1795](https://github.com/flashinfer-ai/flashinfer/pull/1795/files#diff-572d7ee5baecada1fd91e892d5250df6f9a90bf01e9a38c755bb6bacd2f6dd05L37)) has changed the module path of `gen_pod_module`. However, this was not reflected in `test_pod_kernels.py`. Current PR rectifies the import path. Error message Before: ``` ImportError while importing test module '/flashinfer/tests/utils/test_pod_kernels.py'. Hint: make sure your test modules/packages have valid Python names. Traceback: /opt/conda/envs/py312/lib/python3.12/importlib/__init__.py:90: in import_module return _bootstrap._gcd_import(name[level:], package, level) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ tests/utils/test_pod_kernels.py:25: in <module> from flashinfer.jit.attention.pytorch import gen_pod_module E ModuleNotFoundError: No module named 'flashinfer.jit.attention.pytorch' ... ... 1 error in 0.20s ``` After: ``` 168 passed, 24 skipped in 133.16s (0:02:13) ``` <!-- What does this PR do? Briefly describe the changes and why they’re needed. --> ## 🔍 Related Issues <!-- Link any related issues here --> ## 🚀 Pull Request Checklist Thank you for contributing to FlashInfer! Before we review your pull request, please make sure the following items are complete. ### ✅ Pre-commit Checks - [x] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [x] I have installed the hooks with `pre-commit install`. - [x] I have run the hooks manually with `pre-commit run --all-files` and fixed any reported issues. > If you are unsure about how to set up `pre-commit`, see [the pre-commit documentation](https://pre-commit.com/). ## 🧪 Tests - [x] Tests have been added or updated as needed. - [x] All tests are passing (`unittest`, etc.). ## Reviewer Notes <!-- Optional: anything you'd like reviewers to focus on, concerns, etc. -->
1 parent 52089b5 commit 398dc26

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/utils/test_pod_kernels.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
)
2323

2424
import flashinfer
25-
from flashinfer.jit.attention.pytorch import gen_pod_module
25+
from flashinfer.jit.attention import gen_pod_module
2626

2727

2828
@pytest.fixture(autouse=True, scope="module")

0 commit comments

Comments
 (0)