Fix tf text class (#7724)

* Fix test * fix generic text classification * fix test * Fix tests

Fix tf text class (#7724)
* Fix test * fix generic text classification * fix test * Fix tests
d9ffb87e · Julien Plu · GitHub · d6175a42 · d9ffb87e
Unverified Commit d9ffb87e authored Oct 12, 2020 by Julien Plu Committed by GitHub Oct 12, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/text-classification/run_tf_text_classification.py examples/text-classification/run_tf_text_classification.py +1 -1

No files found.
--- 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",