start some annotaition fixes for pytype

3 files changed, 11 insertions, 6 deletions

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)