fix alembic issues

shared/workspaces/alembic/versions/7ec92e7da566_.py

+"""empty message
+
+Revision ID: 7ec92e7da566
+Revises: e680d469ec4e, ce2cca9f1220
+Create Date: 2024-03-20 09:54:03.401572
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = "7ec92e7da566"
+down_revision = ("e680d469ec4e", "ce2cca9f1220")
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    pass
+
+
+def downgrade():
+    pass

shared/workspaces/alembic/versions/ce2cca9f1220_vlass_image_flagtemplate.py

@@ -7,15 +7,13 @@ Create Date: 2024-03-11 10:35:18.654783
 """
 from alembic import op
 from pathlib import Path
-import sqlalchemy as sa
 
 # revision identifiers, used by Alembic.
-revision = 'ce2cca9f1220'
-down_revision = 'd992b379d8a7'
+revision = "ce2cca9f1220"
+down_revision = "d992b379d8a7"
 branch_labels = None
 depends_on = None
 
-conn = op.get_bind()
 old_condor = "envoy_condor_2.8.2.txt"
 new_condor = "envoy_condor_2.8.3.txt"
 
@@ -25,6 +23,8 @@ def set_wf_content(wf_name: str, filename: str) -> bytes:
 
 
 def upgrade():
+    conn = op.get_bind()
+
     # Transfer the image flag template into condor
     conn.execute(
         f"""
@@ -55,6 +55,8 @@ def upgrade():
 
 
 def downgrade():
+    conn = op.get_bind()
+
     conn.execute(
         f"""
          UPDATE workflow_templates