9.4 KiB
How to contribute to transformers?
Everyone is welcome to contribute, and we value everybody's contribution. Code is thus not the only way to help the community. Answering questions, helping others, reaching out and improving the documentations are immensely valuable to the community.
It also helps us if you spread the word: reference the library from blog posts on the awesome projects it made possible, shout out on Twitter every time it has helped you, or simply star the repo to say "thank you".
You can contribute in so many ways!
There are 4 ways you can contribute to transformers:
- Fixing outstanding issues with the existing code;
- Implementing new models;
- Contributing to the examples or to the documentation;
- Submitting issues related to bugs or desired new features.
All are equally valuable to the community.
Submitting a new issue or feature request
Do your best to follow these guidelines when submitting an issue or a feature request. It will make it easier for us to come back to you quickly and with good feedback.
Did you find a bug?
The transformers are robust and reliable thanks to the users who notify us of the problems they encounter. So thank you for reporting an issue.
First, we would really appreciate it if you could make sure the bug was not already reported (use the search bar on Github under Issues).
Did not find it? :( So we can act quickly on it, please follow these steps:
- Include your OS type and version, the versions of Python, PyTorch and Tensorflow when applicable;
- A short, self-contained, code snippet that allows us to reproduce the bug in less than 30s;
- Provide the full traceback if an exception is raised.
To get the OS and software versions automatically, you can run the following command:
python transformers-cli env
Do you want to implement a new model?
Awesome! Please provide the following information:
- Short description of the model and link to the paper;
- Link to the implementation if it is open-source;
- Link to the model weights if they are available.
If you are willing to contribute the model yourself, let us know so we can best guide you.
We have added a detailed guide and templates to guide you in the process of adding a new model. You can find them in the templates
folder.
Do you want a new feature (that is not a model)?
A world-class feature request addresses the following points:
- Motivation first:
- Is it related to a problem/frustration with the library? If so, please explain why. Providing a code snippet that demonstrates the problem is best.
- Is it related to something you would need for a project? We'd love to hear about it!
- Is it something you worked on and think could benefit the community? Awesome! Tell us what problem it solved for you.
- Write a full paragraph describing the feature;
- Provide a code snippet that demonstrates its future use;
- In case this is related to a paper, please attach a link;
- Attach any additional information (drawings, screenshots, etc.) you think may help.
If your issue is well written we're already 80% of the way there by the time you post it.
We have added templates to guide you in the process of adding a new example script for training or testing the models in the library. You can find them in the templates
folder.
Start contributing! (Pull Requests)
Before writing code, we strongly advise you to search through the exising PRs or issues to make sure that nobody is already working on the same thing. If you are unsure, it is always a good idea to open an issue to get some feedback.
You will need basic git
proficiency to be able to contribute to
transformers
. git
is not the easiest tool to use but it has the greatest
manual. Type git --help
in a shell and enjoy. If you prefer books, Pro
Git is a very good reference.
Follow these steps to start contributing:
-
Fork the repository by clicking on the 'Fork' button on the repository's page. This creates a copy of the code under your GitHub user account.
-
Clone your fork to your local disk, and add the base repository as a remote:
$ git clone git@github.com:<your Github handle>/transformers.git $ cd transformers $ git remote add upstream https://github.com/huggingface/transformers.git
-
Create a new branch to hold your development changes:
$ git checkout -b a-descriptive-name-for-my-changes
do not work on the
master
branch. -
Set up a development environment by running the following command in a virtual environment:
$ pip install -e ".[dev]"
(If transformers was already installed in the virtual environment, remove it with
pip uninstall transformers
before reinstalling it in editable mode with the-e
flag.)Right now, we need an unreleased version of
isort
to avoid a bug:$ pip install -U git+git://github.com/timothycrosley/isort.git@e63ae06ec7d70b06df9e528357650281a3d3ec22#egg=isort
-
Develop the features on your branch.
As you work on the features, you should make sure that the test suite passes:
$ make test
transformers
relies onblack
andisort
to format its source code consistently. After you make changes, format them with:$ make style
transformers
also usesflake8
to check for coding mistakes. Quality control runs in CI, however you can also run the same checks with:$ make quality
Once you're happy with your changes, add changed files using
git add
and make a commit withgit commit
to record your changes locally:$ git add modified_file.py $ git commit
Please write good commit messages.
It is a good idea to sync your copy of the code with the original repository regularly. This way you can quickly account for changes:
$ git fetch upstream $ git rebase upstream/master
Push the changes to your account using:
$ git push -u origin a-descriptive-name-for-my-changes
-
Once you are satisfied (and the checklist below is happy too), go to the webpage of your fork on GitHub. Click on 'Pull request' to send your changes to the project maintainers for review.
-
It's ok if maintainers ask you for changes. It happens to core contributors too! So everyone can see the changes in the Pull request, work in your local branch and push the changes to your fork. They will automatically appear in the pull request.
Checklist
- The title of your pull request should be a summary of its contribution;
- If your pull request adresses an issue, please mention the issue number in the pull request description to make sure they are linked (and people consulting the issue know you are working on it);
- To indicate a work in progress please prefix the title with
[WIP]
. These are useful to avoid duplicated work, and to differentiate it from PRs ready to be merged; - Make sure existing tests pass;
- Add high-coverage tests. No quality test, no merge.
- If you are adding a new model, make sure that you use
ModelTester.all_model_classes = (MyModel, MyModelWithLMHead,...)
, which triggers the common tests. - If you are adding new
@slow
tests, make sure they pass usingRUN_SLOW=1 python -m pytest tests/test_my_new_model.py
. CircleCI does not run them.
- All public methods must have informative docstrings;
Tests
You can run 🤗 Transformers tests with unittest
or pytest
.
We like pytest
and pytest-xdist
because it's faster. From the root of the
repository, here's how to run tests with pytest
for the library:
$ python -m pytest -n auto --dist=loadfile -s -v ./tests/
and for the examples:
$ pip install -r examples/requirements.txt # only needed the first time
$ python -m pytest -n auto --dist=loadfile -s -v ./examples/
In fact, that's how make test
and make test-examples
are implemented!
You can specify a smaller set of tests in order to test only the feature you're working on.
By default, slow tests are skipped. Set the RUN_SLOW
environment variable to
yes
to run them. This will download many gigabytes of models — make sure you
have enough disk space and a good Internet connection, or a lot of patience!
$ RUN_SLOW=yes python -m pytest -n auto --dist=loadfile -s -v ./tests/
$ RUN_SLOW=yes python -m pytest -n auto --dist=loadfile -s -v ./examples/
Likewise, set the RUN_CUSTOM_TOKENIZERS
environment variable to yes
to run
tests for custom tokenizers, which don't run by default either.
🤗 Transformers uses pytest
as a test runner only. It doesn't use any
pytest
-specific features in the test suite itself.
This means unittest
is fully supported. Here's how to run tests with
unittest
:
$ python -m unittest discover -s tests -t . -v
$ python -m unittest discover -s examples -t examples -v
Style guide
For documentation strings, transformers
follows the google
style.