File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -118,7 +118,8 @@ We ask contributors to implement the unit tests necessary to check the correctne
118118implementations. Besides, whenever usage examples are provided, we require the example' s code to be
119119tested in `tests/doc`. We require a very high code coverage for newly introduced sources (~95-100%).
120120To ensure that the tensors generated during the tests are on the right device, you have to use the
121- partial functions defined in `tests/unit/_utils.py` to instantiate tensors. For instance, instead of
121+ partial functions defined in `tests/utils/tensors.py` to instantiate tensors. For instance, instead
122+ of
122123```python
123124import torch
124125a = torch.ones(3, 4)
@@ -130,7 +131,7 @@ a = ones_(3, 4)
130131```
131132
132133This will automatically call `torch.ones` with `device=unit.conftest.DEVICE`.
133- If the function you need does not exist yet as a partial function in `_utils .py`, add it.
134+ If the function you need does not exist yet as a partial function in `tensors .py`, add it.
134135Lastly, when you create a model or a random generator, you have to move them manually to the right
135136device (the `DEVICE` defined in `unit.conftest`):
136137```python
You can’t perform that action at this time.
0 commit comments