Fix unit test model config, we need to use kwargs to make dataclass correct.

PiperOrigin-RevId: 327139660

Fix unit test model config, we need to use kwargs to make dataclass correct.
PiperOrigin-RevId: 327139660
853d1a8b · Hongkun Yu · A. Unique TensorFlower · 093f3e1e · 853d1a8b
Commit 853d1a8b authored Aug 17, 2020 by Hongkun Yu Committed by A. Unique TensorFlower Aug 17, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/nlp/tasks/masked_lm_test.py official/nlp/tasks/masked_lm_test.py +1 -1

No files found.
--- a/official/nlp/tasks/masked_lm_test.py
+++ b/official/nlp/tasks/masked_lm_test.py
@@ -29,7 +29,7 @@ class MLMTaskTest(tf.test.TestCase):
    config = masked_lm.MaskedLMConfig(
        init_checkpoint=self.get_temp_dir(),
        model=bert.PretrainerConfig(
-            encoders.EncoderConfig(
+            encoder=encoders.EncoderConfig(
                bert=encoders.BertEncoderConfig(vocab_size=30522,
                                                num_layers=1)),
            cls_heads=[