CONTRIBUTING.md 18.4 KB
Newer Older
Sylvain Gugger's avatar
Sylvain Gugger committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
<!---
Copyright 2020 The HuggingFace Team. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
-->

17
# Contribute to 🤗 Transformers
18
19

Everyone is welcome to contribute, and we value everybody's contribution. Code
20
21
contributions are not the only way to help the community. Answering questions, helping
others, and improving the documentation are also immensely valuable.
22

23
24
25
It also helps us if you spread the word! Reference the library in blog posts
about the awesome projects it made possible, shout out on Twitter every time it has
helped you, or simply ⭐️ the repository to say thank you.
26

27
However you choose to contribute, please be mindful and respect our
28
[code of conduct](https://github.com/huggingface/transformers/blob/main/CODE_OF_CONDUCT.md).
Sylvain Gugger's avatar
Sylvain Gugger committed
29

30
31
32
33
34
**This guide was heavily inspired by the awesome [scikit-learn guide to contributing](https://github.com/scikit-learn/scikit-learn/blob/main/CONTRIBUTING.md).**

## Ways to contribute

There are several ways you can contribute to 🤗 Transformers:
35

36
37
38
39
* Fix outstanding issues with the existing code.
* Submit issues related to bugs or desired new features.
* Implement new models.
* Contribute to the examples or to the documentation.
40

41
If you don't know where to start, there is a special [Good First
42
Issue](https://github.com/huggingface/transformers/contribute) listing. It will give you a list of
43
open issues that are beginner-friendly and help you start contributing to open-source. The best way to do that is to open a Pull Request and link it to the issue that you'd like to work on. We try to give priority to opened PRs as we can easily track the progress of the fix, and if the contributor does not have time anymore, someone else can take the PR over.
44
45
46
47
48
49

For something slightly more challenging, you can also take a look at the [Good Second Issue](https://github.com/huggingface/transformers/labels/Good%20Second%20Issue) list. In general though, if you feel like you know what you're doing, go for it and we'll help you get there! 🚀

> All contributions are equally valuable to the community. 🥰

## Fixing outstanding issues
50

51
If you notice an issue with the existing code and have a fix in mind, feel free to [start contributing](#create-a-pull-request) and open a Pull Request!
52

53
## Submitting a bug-related issue or feature request
54

55
Do your best to follow these guidelines when submitting a bug-related issue or a feature
56
57
58
59
60
request. It will make it easier for us to come back to you quickly and with good
feedback.

### Did you find a bug?

61
The 🤗 Transformers library is robust and reliable thanks to users who report the problems they encounter.
62

63
Before you report an issue, we would really appreciate it if you could **make sure the bug was not
Joao Gante's avatar
Joao Gante committed
64
65
66
67
already reported** (use the search bar on GitHub under Issues). Your issue should also be related to bugs in the library itself, and not your code. If you're unsure whether the bug is in your code or the library, please ask in the [forum](https://discuss.huggingface.co/) or on our [discord](https://discord.com/invite/hugging-face-879548962464493619) first. This helps us respond quicker to fixing issues related to the library versus general questions.

> [!TIP]
> We have a [docs bot](https://huggingface.co/spaces/huggingchat/hf-docs-chat), and we highly encourage you to ask all your questions there. There is always a chance your bug can be fixed with a simple flag 👾🔫
68

69
Once you've confirmed the bug hasn't already been reported, please include the following information in your issue so we can quickly resolve it:
70

71
72
* Your **OS type and version** and **Python**, **PyTorch** and
  **TensorFlow** versions when applicable.
73
* A short, self-contained, code snippet that allows us to reproduce the bug in
74
75
76
  less than 30s.
* The *full* traceback if an exception is raised.
* Attach any other additional information, like screenshots, you think may help.
77

78
To get the OS and software versions automatically, run the following command:
79

80
```bash
81
transformers-cli env
82
83
```

84
You can also run the same command from the root of the repository:
85
86
87
88
89

```bash
python src/transformers/commands/transformers_cli.py env
```

90
### Do you want a new feature?
91

92
If there is a new feature you'd like to see in 🤗 Transformers, please open an issue and describe:
93

94
1. What is the *motivation* behind this feature? Is it related to a problem or frustration with the library? Is it a feature related to something you need for a project? Is it something you worked on and think it could benefit the community?
95

96
97
98
99
100
101
102
   Whatever it is, we'd love to hear about it!

2. Describe your requested feature in as much detail as possible. The more you can tell us about it, the better we'll be able to help you.
3. Provide a *code snippet* that demonstrates the features usage.
4. If the feature is related to a paper, please include a link.

If your issue is well written we're already 80% of the way there by the time you create it.
103

104
We have added [templates](https://github.com/huggingface/transformers/tree/main/templates) to help you get started with your issue.
105

106
## Do you want to implement a new model?
107

108
New models are constantly released and if you want to implement a new model, please provide the following information:
109

110
* A short description of the model and a link to the paper.
111
112
* Link to the implementation if it is open-sourced.
* Link to the model weights if they are available.
113

114
If you are willing to contribute the model yourself, let us know so we can help you add it to 🤗 Transformers!
115

116
We have a technical guide for [how to add a model to 🤗 Transformers](https://huggingface.co/docs/transformers/add_new_model).
117

118
## Do you want to add documentation?
119

120
We're always looking for improvements to the documentation that make it more clear and accurate. Please let us know how the documentation can be improved such as typos and any content that is missing, unclear or inaccurate. We'll be happy to make the changes or help you make a contribution if you're interested!
121

122
For more details about how to generate, build, and write the documentation, take a look at the documentation [README](https://github.com/huggingface/transformers/tree/main/docs).
Rémi Louf's avatar
Rémi Louf committed
123

124
125
126
127
## Create a Pull Request

Before writing any code, we strongly advise you to search through the existing PRs or
issues to make sure nobody is already working on the same thing. If you are
128
unsure, it is always a good idea to open an issue to get some feedback.
Rémi Louf's avatar
Rémi Louf committed
129

130
131
132
You will need basic `git` proficiency to contribute to
🤗 Transformers. While `git` is not the easiest tool to use, it has the greatest
manual. Type `git --help` in a shell and enjoy! If you prefer books, [Pro
Rémi Louf's avatar
Rémi Louf committed
133
134
Git](https://git-scm.com/book/en/v2) is a very good reference.

135
You'll need **[Python 3.8](https://github.com/huggingface/transformers/blob/main/setup.py#L426)** or above to contribute to 🤗 Transformers. Follow the steps below to start contributing:
Rémi Louf's avatar
Rémi Louf committed
136
137

1. Fork the [repository](https://github.com/huggingface/transformers) by
138
   clicking on the **[Fork](https://github.com/huggingface/transformers/fork)** button on the repository's page. This creates a copy of the code
139
140
   under your GitHub user account.

Rémi Louf's avatar
Rémi Louf committed
141
2. Clone your fork to your local disk, and add the base repository as a remote:
142

Rémi Louf's avatar
Rémi Louf committed
143
   ```bash
144
145
146
   git clone git@github.com:<your Github handle>/transformers.git
   cd transformers
   git remote add upstream https://github.com/huggingface/transformers.git
Rémi Louf's avatar
Rémi Louf committed
147
148
149
150
151
   ```

3. Create a new branch to hold your development changes:

   ```bash
152
   git checkout -b a-descriptive-name-for-my-changes
Rémi Louf's avatar
Rémi Louf committed
153
   ```
154

155
   🚨 **Do not** work on the `main` branch!
156

157
4. Set up a development environment by running the following command in a virtual environment:
Rémi Louf's avatar
Rémi Louf committed
158
159

   ```bash
160
   pip install -e ".[dev]"
Rémi Louf's avatar
Rémi Louf committed
161
162
   ```

163
   If 🤗 Transformers was already installed in the virtual environment, remove
164
   it with `pip uninstall transformers` before reinstalling it in editable
165
   mode with the `-e` flag.
Joao Gante's avatar
Joao Gante committed
166

167
168
169
170
   Depending on your OS, and since the number of optional dependencies of Transformers is growing, you might get a
   failure with this command. If that's the case make sure to install the Deep Learning framework you are working with
   (PyTorch, TensorFlow and/or Flax) then do:

171
   ```bash
172
   pip install -e ".[quality]"
173
   ```
Sylvain Gugger's avatar
Sylvain Gugger committed
174

175
176
   which should be enough for most use cases.

177
5. Develop the features in your branch.
178

179
180
   As you work on your code, you should make sure the test suite
   passes. Run the tests impacted by your changes like this:
181
182

   ```bash
183
   pytest tests/<TEST_TO_RUN>.py
184
185
186
   ```

   For more information about tests, check out the
187
   [Testing](https://huggingface.co/docs/transformers/testing) guide.
188

Sylvain Gugger's avatar
Sylvain Gugger committed
189
   🤗 Transformers relies on `black` and `ruff` to format its source code
190
191
   consistently. After you make changes, apply automatic style corrections and code verifications
   that can't be automated in one go with:
Stas Bekman's avatar
Stas Bekman committed
192
193

   ```bash
194
   make fixup
Stas Bekman's avatar
Stas Bekman committed
195
196
   ```

197
   This target is also optimized to only work with files modified by the PR you're working on.
Stas Bekman's avatar
Stas Bekman committed
198

199
   If you prefer to run the checks one after the other, the following command applies the
200
   style corrections:
201
202

   ```bash
203
   make style
204
205
   ```

Sylvain Gugger's avatar
Sylvain Gugger committed
206
   🤗 Transformers also uses `ruff` and a few custom scripts to check for coding mistakes. Quality
207
   controls are run by the CI, but you can run the same checks with:
208
209

   ```bash
210
   make quality
211
   ```
212

213
   Finally, we have a lot of scripts to make sure we don't forget to update
214
   some files when adding a new model. You can run these scripts with:
215
216

   ```bash
217
   make repo-consistency
218
219
   ```

220
221
   To learn more about those checks and how to fix any issues with them, check out the
   [Checks on a Pull Request](https://huggingface.co/docs/transformers/pr_checks) guide.
222

223
   If you're modifying documents under the `docs/source` directory, make sure the documentation can still be built. This check will also run in the CI when you open a pull request. To run a local check
224
   make sure you install the documentation builder:
Joao Gante's avatar
Joao Gante committed
225

226
   ```bash
227
   pip install ".[docs]"
228
229
   ```

230
   Run the following command from the root of the repository:
231
232

   ```bash
233
   doc-builder build transformers docs/source/en --build_dir ~/tmp/test-build
234
235
   ```

236
   This will build the documentation in the `~/tmp/test-build` folder where you can inspect the generated
237
   Markdown files with your favorite editor. You can also preview the docs on GitHub when you open a pull request.
238

239
   Once you're happy with your changes, add the changed files with `git add` and
240
   record your changes locally with `git commit`:
241

Rémi Louf's avatar
Rémi Louf committed
242
   ```bash
243
244
   git add modified_file.py
   git commit
Rémi Louf's avatar
Rémi Louf committed
245
   ```
246

247
248
   Please remember to write [good commit
   messages](https://chris.beams.io/posts/git-commit/) to clearly communicate the changes you made!
249

250
251
   To keep your copy of the code up to date with the original
   repository, rebase your branch on `upstream/branch` *before* you open a pull request or if requested by a maintainer:
252

Rémi Louf's avatar
Rémi Louf committed
253
   ```bash
254
255
   git fetch upstream
   git rebase upstream/main
Rémi Louf's avatar
Rémi Louf committed
256
   ```
257

258
   Push your changes to your branch:
259

Rémi Louf's avatar
Rémi Louf committed
260
   ```bash
261
   git push -u origin a-descriptive-name-for-my-changes
Rémi Louf's avatar
Rémi Louf committed
262
   ```
263

264
265
   If you've already opened a pull request, you'll need to force push with the `--force` flag. Otherwise, if the pull request hasn't been opened yet, you can just push your changes normally.

266
6. Now you can go to your fork of the repository on GitHub and click on **Pull Request** to open a pull request. Make sure you tick off all the boxes on our [checklist](#pull-request-checklist) below. When you're ready, you can send your changes to the project maintainers for review.
267

268
269
7. It's ok if maintainers request changes, it happens to our core contributors
   too! So everyone can see the changes in the pull request, work in your local
270
271
   branch and push the changes to your fork. They will automatically appear in
   the pull request.
Rémi Louf's avatar
Rémi Louf committed
272

273
274
275
276
277
278
279
### Pull request checklist

☐ The pull request title should summarize your contribution.<br>
☐ If your pull request addresses an issue, please mention the issue number in the pull
request description to make sure they are linked (and people viewing the issue know you
are working on it).<br>
☐ To indicate a work in progress please prefix the title with `[WIP]`. These are
280
useful to avoid duplicated work, and to differentiate it from PRs ready to be merged.<br>
281
282
283
284
☐ Make sure existing tests pass.<br>
☐ If adding a new feature, also add tests for it.<br>
   - If you are adding a new model, make sure you use
     `ModelTester.all_model_classes = (MyModel, MyModelWithLMHead,...)` to trigger the common tests.
Stas Bekman's avatar
Stas Bekman committed
285
   - If you are adding new `@slow` tests, make sure they pass using
286
287
288
     `RUN_SLOW=1 python -m pytest tests/models/my_new_model/test_my_new_model.py`.
   - If you are adding a new tokenizer, write tests and make sure
     `RUN_SLOW=1 python -m pytest tests/models/{your_model_name}/test_tokenization_{your_model_name}.py` passes.
289
   - CircleCI does not run the slow tests, but GitHub Actions does every night!<br>
290
291
292
293
294
295
296
297
298
299

☐ All public methods must have informative docstrings (see
[`modeling_bert.py`](https://github.com/huggingface/transformers/blob/main/src/transformers/models/bert/modeling_bert.py)
for an example).<br>
☐ Due to the rapidly growing repository, don't add any images, videos and other
non-text files that'll significantly weigh down the repository. Instead, use a Hub
repository such as [`hf-internal-testing`](https://huggingface.co/hf-internal-testing)
to host these files and reference them by URL. We recommend placing documentation
related images in the following repository:
[huggingface/documentation-images](https://huggingface.co/datasets/huggingface/documentation-images).
300
You can open a PR on this dataset repository and ask a Hugging Face member to merge it.
301

302
For more information about the checks run on a pull request, take a look at our [Checks on a Pull Request](https://huggingface.co/docs/transformers/pr_checks) guide.
Sylvain Gugger's avatar
Sylvain Gugger committed
303

304
305
### Tests

Stas Bekman's avatar
Stas Bekman committed
306
An extensive test suite is included to test the library behavior and several examples. Library tests can be found in
307
308
the [tests](https://github.com/huggingface/transformers/tree/main/tests) folder and examples tests in the
[examples](https://github.com/huggingface/transformers/tree/main/examples) folder.
309
310

We like `pytest` and `pytest-xdist` because it's faster. From the root of the
311
repository, specify a *path to a subfolder or a test file* to run the test:
312
313

```bash
314
python -m pytest -n auto --dist=loadfile -s -v ./tests/models/my_new_model
315
316
```

317
Similarly, for the `examples` directory, specify a *path to a subfolder or test file* to run the test. For example, the following command tests the text classification subfolder in the PyTorch `examples` directory:
318
319

```bash
320
321
pip install -r examples/xxx/requirements.txt  # only needed the first time
python -m pytest -n auto --dist=loadfile -s -v ./examples/pytorch/text-classification
322
323
```

324
325
326
In fact, this is actually how our `make test` and `make test-examples` commands are implemented (not including the `pip install`)!

You can also specify a smaller set of tests in order to test only the feature
327
328
you're working on.

329
330
331
332
333
334
335
336
337
By default, slow tests are skipped but you can set the `RUN_SLOW` environment variable to
`yes` to run them. This will download many gigabytes of models so make sure you
have enough disk space, a good internet connection or a lot of patience!

<Tip warning={true}>

Remember to specify a *path to a subfolder or a test file* to run the test. Otherwise, you'll run all the tests in the `tests` or `examples` folder, which will take a very long time!

</Tip>
338
339

```bash
340
341
RUN_SLOW=yes python -m pytest -n auto --dist=loadfile -s -v ./tests/models/my_new_model
RUN_SLOW=yes python -m pytest -n auto --dist=loadfile -s -v ./examples/pytorch/text-classification
342
343
```

344
345
346
347
348
349
Like the slow tests, there are other environment variables available which not enabled by default during testing:
- `RUN_CUSTOM_TOKENIZERS`: Enables tests for custom tokenizers.
- `RUN_PT_FLAX_CROSS_TESTS`: Enables tests for PyTorch + Flax integration.
- `RUN_PT_TF_CROSS_TESTS`: Enables tests for TensorFlow + PyTorch integration.

More environment variables and additional information can be found in the [testing_utils.py](src/transformers/testing_utils.py).
350
351
352
353
354
355
356
357

🤗 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`:

```bash
358
359
python -m unittest discover -s tests -t . -v
python -m unittest discover -s examples -t examples -v
360
361
```

Rémi Louf's avatar
Rémi Louf committed
362
363
### Style guide

364
For documentation strings, 🤗 Transformers follows the [Google Python Style Guide](https://google.github.io/styleguide/pyguide.html).
365
Check our [documentation writing guide](https://github.com/huggingface/transformers/tree/main/docs#writing-documentation---specification)
366
for more information.
Rémi Louf's avatar
Rémi Louf committed
367

368
369
### Develop on Windows

370
On Windows (unless you're working in [Windows Subsystem for Linux](https://learn.microsoft.com/en-us/windows/wsl/) or WSL), you need to configure git to transform Windows `CRLF` line endings to Linux `LF` line endings:
371

372
373
374
```bash
git config core.autocrlf input
```
375

376
One way to run the `make` command on Windows is with MSYS2:
377

378
379
380
1. [Download MSYS2](https://www.msys2.org/), and we assume it's installed in `C:\msys64`.
2. Open the command line `C:\msys64\msys2.exe` (it should be available from the **Start** menu).
3. Run in the shell: `pacman -Syu` and install `make` with `pacman -S make`.
381
382
4. Add `C:\msys64\usr\bin` to your PATH environment variable.

383
You can now use `make` from any terminal (PowerShell, cmd.exe, etc.)! 🎉
384
385

### Sync a forked repository with upstream main (the Hugging Face repository)
386

387
When updating the main branch of a forked repository, please follow these steps to avoid pinging the upstream repository which adds reference notes to each upstream PR, and sends unnecessary notifications to the developers involved in these PRs.
388

389
1. When possible, avoid syncing with the upstream using a branch and PR on the forked repository. Instead, merge directly into the forked main.
390
2. If a PR is absolutely necessary, use the following steps after checking out your branch:
391

392
393
394
395
396
397
   ```bash
   git checkout -b your-branch-for-syncing
   git pull --squash --no-commit upstream main
   git commit -m '<your message without GitHub references>'
   git push --set-upstream origin your-branch-for-syncing
   ```