Skip to content

Commit 85fde09

Browse files
authored
[pytest] Avoid flash attn test marker warning (#27509)
add flash attn markers
1 parent 1394e08 commit 85fde09

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pyproject.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,3 +21,6 @@ known-first-party = ["transformers"]
2121
[tool.pytest.ini_options]
2222
doctest_optionflags="NUMBER NORMALIZE_WHITESPACE ELLIPSIS"
2323
doctest_glob="**/*.md"
24+
markers = [
25+
"flash_attn_test: marks tests related to flash attention (deselect with '-m \"not flash_attn_test\"')",
26+
]

0 commit comments

Comments
 (0)