Skip to content

Commit

Permalink
Fix CI
Browse files Browse the repository at this point in the history
  • Loading branch information
sgugger committed Aug 20, 2020
1 parent 33bf426 commit b3e5469
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions src/transformers/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -704,9 +704,7 @@ def log(self, logs: Dict[str, float], iterator: Optional[tqdm] = None) -> None:
else:
print(output)

def _prepare_inputs(
self, inputs: Dict[str, Union[torch.Tensor, Any]]
) -> Dict[str, Union[torch.Tensor, Any]]:
def _prepare_inputs(self, inputs: Dict[str, Union[torch.Tensor, Any]]) -> Dict[str, Union[torch.Tensor, Any]]:
"""
Prepare :obj:`inputs` before feeding them to the model, converting them to tensors if they are not already and
handling potential state.
Expand Down

0 comments on commit b3e5469

Please sign in to comment.