From d9ffb87efb939cec920abd10a5843906d81b1815 Mon Sep 17 00:00:00 2001 From: Julien Plu Date: Mon, 12 Oct 2020 14:45:15 +0200 Subject: [PATCH] Fix tf text class (#7724) * Fix test * fix generic text classification * fix test * Fix tests --- examples/text-classification/run_tf_text_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/text-classification/run_tf_text_classification.py b/examples/text-classification/run_tf_text_classification.py index 657119abb8..cb3b75da7b 100644 --- a/examples/text-classification/run_tf_text_classification.py +++ b/examples/text-classification/run_tf_text_classification.py @@ -60,7 +60,7 @@ def get_tfds( for k in files.keys(): transformed_ds[k] = ds[k].map( lambda example: tokenizer.batch_encode_plus( - (example[features_name[0]], features_name[1]), + (example[features_name[0]], example[features_name[1]]), truncation=True, max_length=max_seq_length, padding="max_length",