remove any

apache · May 12, 2022 · afa8760 · afa8760
1 parent 9983ce9
commit afa8760
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/superset/dao/datasource/dao.py b/superset/dao/datasource/dao.py
@@ -15,7 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
-from typing import Any, Dict, List, Optional, Set, Union
+from enum import Enum
+from typing import Any, Dict, List, Optional, Set, Type, Union
 
 from flask_babel import _
 from sqlalchemy import or_
@@ -32,12 +33,12 @@
 from superset.tables.models import Table
 from superset.utils.core import DatasourceType
 
-Datasource = Union[Dataset, SqlaTable, Table, Query, SavedQuery, Any]
+Datasource = Union[Dataset, SqlaTable, Table, Query, SavedQuery]
 
 
 class DatasourceDAO(BaseDAO):
 
-    sources: Dict[DatasourceType, Datasource] = {
+    sources: Dict[DatasourceType, Type[Datasource]] = {
         DatasourceType.SQLATABLE: SqlaTable,
         DatasourceType.QUERY: Query,
         DatasourceType.SAVEDQUERY: SavedQuery,