Fix tf text class (#7724)

* Fix test * fix generic text classification * fix test * Fix tests
2020-10-12 14:45:15 +02:00
parent d6175a4268
commit d9ffb87efb
1 changed files with 1 additions and 1 deletions
--- 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",