From aef4e094ab746a717348158f136b8909fc47c0c5 Mon Sep 17 00:00:00 2001 From: Emma Rose Date: Fri, 26 Jul 2019 18:26:14 -0400 Subject: [PATCH] Fix type signature of query.wrapper --- alerts/geomodel/query.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/alerts/geomodel/query.py b/alerts/geomodel/query.py index 79d0c183..26a39d8c 100644 --- a/alerts/geomodel/query.py +++ b/alerts/geomodel/query.py @@ -1,7 +1,7 @@ '''To make GeoModel code more testable, we abstract interaction with ElasticSearch away via a "query interface". This is just a function that, -called with a list of ES indices and a `SearchQuery`, produces a list of -dictionaries as output. +called with an ES index and a `SearchQuery`, produces a list of dictionaries +as output. ''' from typing import Any, Callable, Dict, List @@ -10,7 +10,7 @@ from mozdef_util.elasticsearch_client import ElasticsearchClient as ESClient from mozdef_util.query_models import SearchQuery -QueryInterface = Callable[[SearchQuery, List[str]], List[Dict[str, Any]]] +QueryInterface = Callable[[SearchQuery, str], List[Dict[str, Any]]] def wrap(client: ESClient) -> QueryInterface: