Skip to content

Commit 77a1404

Browse files
committed
fix(test): drop stale model_config kwarg from path_c sparse_mla indent test
_mamba3_fp8_train_acceptance_region was simplified to accept only include_backward; the per-config knobs now flow through the bundled acceptance fixture region builder. The 3 indent-emission tests were still calling with the dropped kwarg and erroring during collection. After: tests/test_path_c_sparse_mla_apply_emit_indent.py 3/3 ✅
1 parent b691440 commit 77a1404

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tests/test_path_c_sparse_mla_apply_emit_indent.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,16 +29,16 @@
2929
import contextlib
3030
import io
3131

32-
from cppmega_mlx.recipes.model_factory import local_gb10_quarter_profile
3332
from cppmega_mlx.runtime import path_c_fusion_schedules as schedules
3433

3534

3635
def _rendered_train_block_source() -> str:
37-
profile = local_gb10_quarter_profile()
38-
tiny = profile.tiny_smoke_config()
36+
# _mamba3_fp8_train_acceptance_region was simplified to take only
37+
# include_backward; the per-config knobs now flow through the
38+
# bundled acceptance fixture region builder.
3939
with contextlib.redirect_stderr(io.StringIO()):
4040
region = schedules._mamba3_fp8_train_acceptance_region(
41-
include_backward=True, model_config=tiny,
41+
include_backward=True,
4242
)
4343
prim_func = schedules.mamba3_fp8_train_fusion_schedule_template(region)
4444
return str(prim_func)

0 commit comments

Comments
 (0)