Correctly run replace_unknown_token code path in test
@@ -177,7 +177,7 @@ class ModelTest(tf.test.TestCase):
model, params = _seq2seq_model()
params["replace_unknown_target"] = True
features_file, labels_file, data_config = self._makeToyEnDeData()
- model.initialize(data_config)
+ model.initialize(data_config, params=params)
dataset = model.examples_inputter.make_inference_dataset(features_file, 16)
features = next(iter(dataset))
_, predictions = model(features)
