From ab923df6a457bf7df8ddaab293299f5d918ec313 Mon Sep 17 00:00:00 2001
From: Tim Ryan <tim.ryan@bluelabs.com>
Date: Mon, 24 Jul 2023 17:19:28 -0400
Subject: [PATCH] RM-96 try to use db-driver

---
 records_mover/records/records.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/records_mover/records/records.py b/records_mover/records/records.py
index 04de3ce7e..e804e075b 100644
--- 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: