From c204c25314d5a92c5095735c4cf217787b1fa921 Mon Sep 17 00:00:00 2001
From: mohamad <mohamad>
Date: Sun, 1 Jun 2025 18:12:58 +0200
Subject: [PATCH] refactor: Modify upgrade function to accept context parameter
 for enhanced migration flexibility

---
 be/alembic/versions/0001_initial_schema.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/be/alembic/versions/0001_initial_schema.py b/be/alembic/versions/0001_initial_schema.py
index ad2aa7f..bf86155 100644
--- a/be/alembic/versions/0001_initial_schema.py
+++ b/be/alembic/versions/0001_initial_schema.py
@@ -25,7 +25,7 @@ recurrence_type_enum = postgresql.ENUM('DAILY', 'WEEKLY', 'MONTHLY', 'YEARLY', n
 chore_frequency_enum = postgresql.ENUM('one_time', 'daily', 'weekly', 'monthly', 'custom', name='chorefrequencyenum', create_type=False)
 chore_type_enum = postgresql.ENUM('personal', 'group', name='choretypeenum', create_type=False)
 
-def upgrade() -> None:
+def upgrade(context) -> None:
     user_role_enum.create(op.get_bind(), checkfirst=True)
     split_type_enum.create(op.get_bind(), checkfirst=True)
     expense_split_status_enum.create(op.get_bind(), checkfirst=True)
-- 
2.45.2