[`pytest`] Avoid flash attn test marker warning (#27509)

add flash attn markers

[`pytest`] Avoid flash attn test marker warning (#27509)
add flash attn markers
85fde09c · Arthur · GitHub · 1394e08c · 85fde09c
Unverified Commit 85fde09c authored Nov 16, 2023 by Arthur Committed by GitHub Nov 16, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

pyproject.toml pyproject.toml +3 -0

No files found.
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,3 +21,6 @@ known-first-party = ["transformers"]
 [tool.pytest.ini_options]
 doctest_optionflags="NUMBER NORMALIZE_WHITESPACE ELLIPSIS"
 doctest_glob="**/*.md"
+markers = [
+    "flash_attn_test: marks tests related to flash attention (deselect with '-m \"not flash_attn_test\"')",
+]
\ No newline at end of file