From dfd233b9856eb8aacfd000f1ac5941beb8514283 Mon Sep 17 00:00:00 2001 From: Weves Date: Sat, 25 May 2024 13:04:09 -0700 Subject: [PATCH] Fix mypy for mediawiki connector --- backend/danswer/connectors/mediawiki/family.py | 4 ++-- backend/danswer/connectors/mediawiki/wiki.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/backend/danswer/connectors/mediawiki/family.py b/backend/danswer/connectors/mediawiki/family.py index 7d0c06d0b..0d9530667 100644 --- a/backend/danswer/connectors/mediawiki/family.py +++ b/backend/danswer/connectors/mediawiki/family.py @@ -9,7 +9,7 @@ from urllib.parse import urlparse from urllib.parse import urlunparse from pywikibot import family # type: ignore[import-untyped] -from pywikibot import pagegenerators +from pywikibot import pagegenerators # type: ignore[import-untyped] from pywikibot.scripts import generate_family_file # type: ignore[import-untyped] from pywikibot.scripts.generate_user_files import pywikibot # type: ignore[import-untyped] @@ -132,7 +132,7 @@ def family_class_dispatch(url: str, name: str) -> type[family.Family]: """ if "wikipedia" in url: - import pywikibot.families.wikipedia_family + import pywikibot.families.wikipedia_family # type: ignore[import-untyped] return pywikibot.families.wikipedia_family.Family # TODO: Support additional families pre-defined in `pywikibot.families.*_family.py` files diff --git a/backend/danswer/connectors/mediawiki/wiki.py b/backend/danswer/connectors/mediawiki/wiki.py index 9bbace174..2283d8130 100644 --- a/backend/danswer/connectors/mediawiki/wiki.py +++ b/backend/danswer/connectors/mediawiki/wiki.py @@ -6,9 +6,9 @@ from collections.abc import Generator from typing import Any from typing import ClassVar -import pywikibot.time -from pywikibot import pagegenerators -from pywikibot import textlib +import pywikibot.time # type: ignore[import-untyped] +from pywikibot import pagegenerators # type: ignore[import-untyped] +from pywikibot import textlib # type: ignore[import-untyped] from danswer.configs.app_configs import INDEX_BATCH_SIZE from danswer.configs.constants import DocumentSource