RM-96 try to use db-driver

bluelabsio · Jul 24, 2023 · ab923df · ab923df
1 parent 0cfc024
commit ab923df
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/records_mover/records/records.py b/records_mover/records/records.py
@@ -60,8 +60,7 @@ class Records:
     "Alias of :meth:`records_mover.records.move`"
 
     def __init__(self,
-                 db_driver: Union[Callable[[Union['Engine', 'Connection']], 'DBDriver'],
-                                  PleaseInfer] = PleaseInfer.token,
+                 db_driver: Union['DBDriver', PleaseInfer] = PleaseInfer.token,
                  url_resolver: Union[UrlResolver, PleaseInfer] = PleaseInfer.token,
                  session: Union['Session', PleaseInfer] = PleaseInfer.token) -> None:
         if db_driver is PleaseInfer.token or url_resolver is PleaseInfer.token: