Fix migration to backfill user.pubid column part 2

hypothesis · Jan 16, 2025 · 344de3c · 344de3c
1 parent a1cf156
commit 344de3c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/h/migrations/versions/f29f80083c26_backfill_user_pubid_part_2.py b/h/migrations/versions/f29f80083c26_backfill_user_pubid_part_2.py
@@ -1,6 +1,7 @@
 """Backfill user.pubid with unique values part 2."""
 
 from alembic import op
+from sqlalchemy.orm import sessionmaker
 
 from h.migrations.versions.f32200e2e496_backfill_user_pubid import backfill_users
 
@@ -14,7 +15,7 @@
 
 
 def upgrade():
-    session = op.get_bind()
+    session = sessionmaker()(bind=op.get_bind())
 
     backfill_users(
         session,