enable result buffering for postgres

1 files changed, 2 insertions, 2 deletions

diff --git a/autoimport.py b/autoimport.py
index 6db6171..7d5c000 100755
--- a/autoimport.py
+++ b/autoimport.py
@@ -18,7 +18,7 @@ from debian import deb822
 from debian.debian_support import version_compare
 import sqlalchemy
 
-from dedup.utils import enable_sqlite_foreign_keys
+from dedup.utils import configure_database_engine
 
 from readyaml import readyaml
 
@@ -91,7 +91,7 @@ def main():
     options, args = parser.parse_args()
     tmpdir = tempfile.mkdtemp(prefix=b"debian-dedup")
     db = sqlalchemy.create_engine(options.database)
-    enable_sqlite_foreign_keys(db)
+    configure_database_engine(db)
     e = concurrent.futures.ThreadPoolExecutor(multiprocessing.cpu_count())
     pkgs = {}
     for d in args: