From d4332643c48545ca158a25bf6e9847a007f04c2f Mon Sep 17 00:00:00 2001 From: H <43509927+guoyuhao2330@users.noreply.github.com> Date: Mon, 15 Jul 2024 19:17:17 +0800 Subject: [PATCH] fix wikipedia language (#1519) ### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- graph/component/wikipedia.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/graph/component/wikipedia.py b/graph/component/wikipedia.py index 48abefaf6..4da911fb8 100644 --- a/graph/component/wikipedia.py +++ b/graph/component/wikipedia.py @@ -30,11 +30,11 @@ class WikipediaParam(ComponentParamBase): def __init__(self): super().__init__() self.top_n = 10 - self.lang = 'en' + self.language = 'en' def check(self): self.check_positive_integer(self.top_n, "Top N") - self.check_valid_value(self.lang, "Wikipedia languages", + self.check_valid_value(self.language, "Wikipedia languages", ['af', 'pl', 'ar', 'ast', 'az', 'bg', 'nan', 'bn', 'be', 'ca', 'cs', 'cy', 'da', 'de', 'et', 'el', 'en', 'es', 'eo', 'eu', 'fa', 'fr', 'gl', 'ko', 'hy', 'hi', 'hr', 'id', 'it', 'he', 'ka', 'lld', 'la', 'lv', 'lt', 'hu', 'mk', 'arz', 'ms', 'min', 'my', 'nl', @@ -52,7 +52,7 @@ class Wikipedia(ComponentBase, ABC): return Wikipedia.be_output(self._param.no) wiki_res = [] - wikipedia.set_lang(self._param.lang) + wikipedia.set_lang(self._param.language) wiki_engine = wikipedia for wiki_key in wiki_engine.search(ans, results=self._param.top_n): try: