From b3e54698dd701667ba1d06501a7a9e431c020863 Mon Sep 17 00:00:00 2001 From: sgugger Date: Thu, 20 Aug 2020 08:34:02 -0400 Subject: [PATCH] Fix CI --- src/transformers/trainer.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index bcba5c89c2..efc29467f1 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -704,9 +704,7 @@ class Trainer: 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.