From 237fc32127a29c11ee881233610c35a9e8252895 Mon Sep 17 00:00:00 2001
From: Jonas Haag <jonas@lophus.org>
Date: Wed, 4 Dec 2024 11:22:29 +0100
Subject: [PATCH] Fix template types

---
 src/pymongo_migrate/generate.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/pymongo_migrate/generate.py b/src/pymongo_migrate/generate.py
index e7f5911..439a835 100644
--- a/src/pymongo_migrate/generate.py
+++ b/src/pymongo_migrate/generate.py
@@ -8,16 +8,17 @@
 {description}
 """
 import pymongo
+import pymongo.database
 
 name = {name!r}
 dependencies = {dependencies!r}
 
 
-def upgrade(db: "pymongo.database.Database"):
+def upgrade(db: pymongo.database.Database):
     pass
 
 
-def downgrade(db: "pymongo.database.Database"):
+def downgrade(db: pymongo.database.Database):
     pass
 '''
 MAX_NAME_LEN = 60