Skip to content

Conversation

@ArthurZucker
Copy link
Collaborator

What does this PR do?

Adds the flash_attn_test marker to the pyproject.toml to remove the following warning:

tests/test_modeling_common.py:3131
  /home/arthur/transformers/tests/test_modeling_common.py:3131: PytestUnknownMarkWarning: Unknown pytest.mark.flash_attn_test - is this a typo?  You can register custom marks to avoid this warning - for details, see https://docs.pytest.org/en/stable/how-to/mark.html
    @mark.flash_attn_test

tests/test_modeling_common.py:3174
  /home/arthur/transformers/tests/test_modeling_common.py:3174: PytestUnknownMarkWarning: Unknown pytest.mark.flash_attn_test - is this a typo?  You can register custom marks to avoid this warning - for details, see https://docs.pytest.org/en/stable/how-to/mark.html
    @mark.flash_attn_test

Allows to de-select them with:

pytest ... -m "not flash_attn_test"

@ArthurZucker ArthurZucker changed the title Avoid-marker-warning [pytest] Avoid-marker-warning Nov 15, 2023
@ArthurZucker ArthurZucker changed the title [pytest] Avoid-marker-warning [pytest] Avoid flash attn test marker warning Nov 15, 2023
Copy link
Member

@LysandreJik LysandreJik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks

@ArthurZucker ArthurZucker merged commit 85fde09 into main Nov 16, 2023
@ArthurZucker ArthurZucker deleted the avoid-marker-warning branch November 16, 2023 10:13
EduardoPach pushed a commit to EduardoPach/transformers that referenced this pull request Nov 19, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants