fix type annotations of bql-related functions (#44)

matthewgilbert · Jul 14, 2024 · b14d8a7 · b14d8a7
1 parent 2369b8e
commit b14d8a7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/blp/blp.py b/src/blp/blp.py
@@ -842,7 +842,7 @@ def bql(
         expression: str,
         overrides: Optional[Sequence] = None,
         options: Optional[Dict] = None,
-    ) -> pandas.DataFrame:
+    ) -> List[pandas.DataFrame]:
         """Bloomberg query language request.
 
         Args:
@@ -874,9 +874,9 @@ def bql(
 
         list_of_dfs = self.query(query, bql_parser, self.collect_to_bql)
 
-        return list_of_dfs
+        return list_of_dfs # type:ignore
 
-    def collect_to_bql(self, responses: Iterable) -> pandas.DataFrame:
+    def collect_to_bql(self, responses: Iterable) -> List[pandas.DataFrame]:
         """Collector for bql()."""
         return [pandas.DataFrame(field) for field in responses]