nvm need to drop alembic table

truera · Dec 4, 2024 · e770f15 · e770f15
1 parent 019799d
commit e770f15
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/core/trulens/core/database/sqlalchemy.py b/src/core/trulens/core/database/sqlalchemy.py
@@ -26,6 +26,7 @@
 import pydantic
 from pydantic import Field
 import sqlalchemy as sa
+from sqlalchemy import Table
 from sqlalchemy.orm import joinedload
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy.sql import text as sql_text
@@ -350,6 +351,8 @@ def reset_database(self):
         meta = self.orm.metadata
 
         tables = [
+            Table(f"{self.table_prefix}alembic_version", self.orm.metadata)
+        ] + [
             c.__table__
             for c in self.orm.registry.values()
             if hasattr(c, "__table__")