more lint

Signed-off-by: benrifkind <ben.rifkind@gmail.com>
amundsen-io · Feb 4, 2021 · 3bcf14b · 3bcf14b
1 parent 538038d
commit 3bcf14b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/extractor/test_sql_alchemy_extractor.py b/tests/unit/extractor/test_sql_alchemy_extractor.py
@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import unittest
-from typing import Any
+from typing import Any, Dict
 
 from mock import patch
 from pyhocon import ConfigFactory
@@ -100,7 +100,7 @@ def test_get_connection(self: Any, mock_method: Any) -> None:
         Test that configs are passed through correctly to the _get_connection method
         """
         extractor = SQLAlchemyExtractor()
-        config_dict = {
+        config_dict: Dict[str, Any] = {
             'extractor.sqlalchemy.conn_string': 'TEST_CONNECTION',
             'extractor.sqlalchemy.extract_sql': 'SELECT 1 FROM TEST_TABLE;'
         }