Skip to content

Commit a65fa03

Browse files
pavlejosipovicPavle Josipovicclaude
committed
Fix broken import in test_deepseek_mla_ops.py after SDPA test migration (#37919)
Regression introduced in 89175b6 (#37713) which moved MLA test files from tests/tt_eager/.../misc/ to tests/ttnn/unit_tests/operations/sdpa/. The file test_deepseek_mla_ops.py was not moved but still imported page_table_setup, to_paged_cache, and from_paged_cache from the old path (test_flash_multi_latent_attention_decode), causing a ModuleNotFoundError during pytest collection and failing all nightly L2 eager unit test jobs. Update the import to point to the new location in mla_test_utils.py. Nightly L2 run wiht fast dispatch unit tests https://github.com/tenstorrent/tt-metal/actions/runs/22032896930 Co-authored-by: Pavle Josipovic <pjosipovic@tenstorrent.com> Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 4ccd355 commit a65fa03

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tt_eager/python_api_testing/unit_testing/misc/test_deepseek_mla_ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
comp_allclose,
1414
comp_pcc,
1515
)
16-
from tests.tt_eager.python_api_testing.unit_testing.misc.test_flash_multi_latent_attention_decode import (
16+
from tests.ttnn.unit_tests.operations.sdpa.mla_test_utils import (
1717
page_table_setup,
1818
to_paged_cache,
1919
from_paged_cache,

0 commit comments

Comments
 (0)