From 9bdcba53fd01309050c0bba7d8803f96c35b343c Mon Sep 17 00:00:00 2001 From: thomwolf Date: Sat, 9 Feb 2019 17:07:12 +0100 Subject: [PATCH] fix tests --- tests/modeling_openai_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/modeling_openai_test.py b/tests/modeling_openai_test.py index 6baaaf677a..1cc8b7d5dc 100644 --- a/tests/modeling_openai_test.py +++ b/tests/modeling_openai_test.py @@ -93,7 +93,7 @@ class OpenAIGPTModelTest(unittest.TestCase): if self.use_labels: mc_labels = OpenAIGPTModelTest.ids_tensor([self.batch_size], self.type_sequence_label_size) lm_labels = OpenAIGPTModelTest.ids_tensor([self.batch_size, self.n_choices, self.seq_length], self.num_labels) - mc_token_ids = OpenAIGPTModelTest.ids_tensor([self.batch_size, self.n_choices], self.seq_length).float() + mc_token_ids = OpenAIGPTModelTest.ids_tensor([self.batch_size, self.n_choices], self.seq_length) config = OpenAIGPTConfig( vocab_size_or_config_json_file=self.vocab_size,