Fix CI

b3e54698 · sgugger · 33bf4264 · b3e54698
Commit b3e54698 authored Aug 20, 2020 by sgugger
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

src/transformers/trainer.py src/transformers/trainer.py +1 -3

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -704,9 +704,7 @@ class Trainer:
        else:
            print(output)
-    def _prepare_inputs(
+    def _prepare_inputs(self, inputs: Dict[str, Union[torch.Tensor, Any]]) -> Dict[str, Union[torch.Tensor, Any]]:
-        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.