Skip to content

Commit

Permalink
Update tests/test_modeling_flaubert.py
Browse files Browse the repository at this point in the history
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
  • Loading branch information
LysandreJik and sgugger committed Jul 29, 2020
1 parent 3203830 commit 42100b3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_modeling_flaubert.py
Original file line number Diff line number Diff line change
Expand Up @@ -328,7 +328,7 @@ def create_and_check_flaubert_token_classif(
self.parent.assertListEqual(list(result["logits"].size()), [self.batch_size, self.seq_length, self.num_labels])
self.check_loss_output(result)

def create_and_check_bert_for_multiple_choice(
def create_and_check_flaubert_for_multiple_choice(
self,
config,
input_ids,
Expand Down

0 comments on commit 42100b3

Please sign in to comment.