From 8556a7515ab6cdd75300014bf48abd7f1147b62d Mon Sep 17 00:00:00 2001
From: michalkrzem <michal_krzem@wp.pl>
Date: Wed, 9 Oct 2024 10:58:56 +0200
Subject: [PATCH] alembic_postgresql_enum for automatic enum type recognize

---
 requirements.txt      | 3 ++-
 src/migrations/env.py | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 305158fe..5ceb863b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -31,4 +31,5 @@ urllib3==2.2.3
 uvicorn==0.30.6
 wrapt==1.16.0
 yara-python==4.5.1
-yaramod==3.23.0
\ No newline at end of file
+yaramod==3.23.0
+alembic_postgresql_enum
\ No newline at end of file
diff --git a/src/migrations/env.py b/src/migrations/env.py
index 68b19a69..a5501150 100644
--- a/src/migrations/env.py
+++ b/src/migrations/env.py
@@ -1,3 +1,4 @@
+import alembic_postgresql_enum # noqa: ignore=F401
 from sqlalchemy import create_engine
 from alembic import context
 from sqlmodel import SQLModel