Fixed pgvector extension not working

agronholm · Feb 22, 2024 · 4ebc666 · 4ebc666
1 parent 3de07dd
commit 4ebc666
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -1,6 +1,10 @@
 Version history
 ===============
 
+**UNRELEASED**
+
+- Fixed pgvector support not working
+
 **3.0.0rc4**
 
 - Dropped support for Python 3.7

diff --git a/src/sqlacodegen/cli.py b/src/sqlacodegen/cli.py
@@ -19,7 +19,7 @@
     geoalchemy2 = None
 
 try:
-    import pgvector
+    import pgvector.sqlalchemy
 except ImportError:
     pgvector = None