huggingface / trl

Train transformer language models with reinforcement learning.
http://hf.co/docs/trl
Apache License 2.0
10.13k stars 1.28k forks source link

🧲 Use our own `require_bitsandbytes` #2370

Closed qgallouedec closed 2 days ago

qgallouedec commented 3 days ago

What does this PR do?

To avoid this warning in our tests:

../transformers/src/transformers/testing_utils.py:1163
  /fsx/qgallouedec/transformers/src/transformers/testing_utils.py:1163: PytestUnknownMarkWarning: Unknown pytest.mark.bitsandbytes - 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
    return pytest.mark.bitsandbytes(test_case)

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag members/contributors who may be interested in your PR.

HuggingFaceDocBuilderDev commented 3 days ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

qgallouedec commented 3 days ago

Failing test not related (fixed in #2373)