From 649913ba9fbab8e9b7b428dec6526797c5b1c532 Mon Sep 17 00:00:00 2001 From: Bryan Newbold Date: Wed, 3 Jun 2020 23:58:38 -0700 Subject: start some annotaition fixes for pytype --- fatcat_scholar/schema.py | 4 +++- fatcat_scholar/search.py | 9 ++++++--- fatcat_scholar/web.py | 4 ++-- 3 files changed, 11 insertions(+), 6 deletions(-) (limited to 'fatcat_scholar') diff --git a/fatcat_scholar/schema.py b/fatcat_scholar/schema.py index 616857e..187fe58 100644 --- a/fatcat_scholar/schema.py +++ b/fatcat_scholar/schema.py @@ -10,8 +10,10 @@ from enum import Enum from typing import Optional, List, Any import ftfy -from pydantic import BaseModel from bs4 import BeautifulSoup +# pytype: disable=import-error +from pydantic import BaseModel +# pytype: enable=import-error from fatcat_openapi_client import ReleaseEntity, ReleaseContrib from fatcat_scholar.api_entities import entity_to_dict diff --git a/fatcat_scholar/search.py b/fatcat_scholar/search.py index 6402e2b..a529f6a 100644 --- a/fatcat_scholar/search.py +++ b/fatcat_scholar/search.py @@ -3,13 +3,16 @@ Helpers to make elasticsearch queries. """ import sys -from gettext import gettext import datetime +from gettext import gettext +from typing import List, Optional, Any + import elasticsearch -from pydantic import BaseModel from dynaconf import settings from elasticsearch_dsl import Search, Q -from typing import List, Optional, Any +# pytype: disable=import-error +from pydantic import BaseModel +# pytype: enable=import-error # i18n note: the use of gettext below doesn't actually do the translation here, # it just ensures that the strings are caught by babel for translation later diff --git a/fatcat_scholar/web.py b/fatcat_scholar/web.py index 4e912d4..8b55f75 100644 --- a/fatcat_scholar/web.py +++ b/fatcat_scholar/web.py @@ -70,14 +70,14 @@ async def search(query: FulltextQuery = Depends(FulltextQuery)) -> Any: web = APIRouter() -def locale_gettext(translations: List[Any]) -> Any: +def locale_gettext(translations: Any) -> Any: def gt(s): return translations.ugettext(s) return gt -def locale_ngettext(translations: List[Any]) -> Any: +def locale_ngettext(translations: Any) -> Any: def ngt(s, n): return translations.ungettext(s) -- cgit v1.2.3