Skip to content

GitLab

Unverified Commit dfbf0f55 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

topk -> top_k (#9541)

parent a1100fac
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_pipelines_fill_mask.py
View file @ dfbf0f55
...@@ -183,7 +183,7 @@ class FillMaskPipelineTests(MonoInputPipelineCommonMixin, unittest.TestCase): ...@@ -183,7 +183,7 @@ class FillMaskPipelineTests(MonoInputPipelineCommonMixin, unittest.TestCase):
] ]
valid_targets = [" Patrick", " Clara"] valid_targets = [" Patrick", " Clara"]
for model_name in self.large_models: 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) mono_result = nlp(valid_inputs[0], targets=valid_targets)
self.assertIsInstance(mono_result, list) self.assertIsInstance(mono_result, list)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment