Change DBInfo return type

tokern · Jun 23, 2023 · 2ebd6f5 · 2ebd6f5
1 parent 2f694a5
commit 2ebd6f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/piicatcher/dbinfo.py b/piicatcher/dbinfo.py
@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import List, Type
+from typing import List
 
 from dbcat.catalog import CatSchema, CatTable
 
@@ -146,7 +146,7 @@ class Athena(Postgres):
     pass
 
 
-def get_dbinfo(source_type: str, *args, **kwargs) -> Type[DbInfo]:
+def get_dbinfo(source_type: str, *args, **kwargs) -> DbInfo:
     if source_type == "sqlite":
         return Sqlite(*args, **kwargs)
     elif source_type == "mysql":

diff --git a/piicatcher/generators.py b/piicatcher/generators.py
@@ -1,7 +1,7 @@
 import datetime
 import logging
 import re
-from typing import Generator, List, Optional, Tuple, Type
+from typing import Generator, List, Optional, Tuple
 
 from dbcat.catalog import Catalog, CatColumn, CatSchema, CatSource, CatTable
 from dbcat.generators import NoMatchesError, table_generator
@@ -44,7 +44,7 @@ def column_generator(
 def _get_table_count(
     schema: CatSchema,
     table: CatTable,
-    dbinfo: Type[DbInfo],
+    dbinfo: DbInfo,
     connection,
     source=CatSource,
 ) -> int:
@@ -61,7 +61,7 @@ def _get_query(
     schema: CatSchema,
     table: CatTable,
     column_list: List[CatColumn],
-    dbinfo: Type[DbInfo],
+    dbinfo: DbInfo,
     connection,
     source: CatSource,
     sample_size: int = SMALL_TABLE_MAX,