From dfbf0f5598700e60e7038ed72cc1d529a0675c9a Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Tue, 12 Jan 2021 16:21:29 -0500 Subject: [PATCH] topk -> top_k (#9541) --- tests/test_pipelines_fill_mask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_pipelines_fill_mask.py b/tests/test_pipelines_fill_mask.py index d4aa79c4ad..f087ed2135 100644 --- a/tests/test_pipelines_fill_mask.py +++ b/tests/test_pipelines_fill_mask.py @@ -183,7 +183,7 @@ class FillMaskPipelineTests(MonoInputPipelineCommonMixin, unittest.TestCase): ] valid_targets = [" Patrick", " Clara"] for model_name in self.large_models: - nlp = pipeline(task="fill-mask", model=model_name, tokenizer=model_name, framework="tf", topk=2) + nlp = pipeline(task="fill-mask", model=model_name, tokenizer=model_name, framework="tf", top_k=2) mono_result = nlp(valid_inputs[0], targets=valid_targets) self.assertIsInstance(mono_result, list)