Removed re-exported classes

Author: ankane

CHANGELOG.md

@@ -1,6 +1,7 @@
 ## 0.5.0 (unreleased)
 
 - Added type hints
+- Removed re-exported classes (use top-level `pgvector` package instead)
 - Dropped support for Python < 3.10
 - Dropped support for SQLAlchemy < 2
 

pgvector/asyncpg/__init__.py

@@ -1,11 +1,5 @@
 from .register import register_vector
 
-# TODO remove
-from .. import Vector, HalfVector, SparseVector
-
 __all__ = [
-    'register_vector',
-    'Vector',
-    'HalfVector',
-    'SparseVector'
+    'register_vector'
 ]

pgvector/django/__init__.py

@@ -6,9 +6,6 @@
 from .sparsevec import SparseVectorField
 from .vector import VectorField
 
-# TODO remove
-from .. import HalfVector, SparseVector
-
 __all__ = [
     'VectorExtension',
     'VectorField',
@@ -22,7 +19,5 @@
     'CosineDistance',
     'L1Distance',
     'HammingDistance',
-    'JaccardDistance',
-    'HalfVector',
-    'SparseVector'
+    'JaccardDistance'
 ]

pgvector/peewee/__init__.py

@@ -3,14 +3,9 @@
 from .sparsevec import SparseVectorField
 from .vector import VectorField
 
-# TODO remove
-from .. import HalfVector, SparseVector
-
 __all__ = [
     'VectorField',
     'HalfVectorField',
     'FixedBitField',
-    'SparseVectorField',
-    'HalfVector',
-    'SparseVector'
+    'SparseVectorField'
 ]

pgvector/psycopg/__init__.py

@@ -1,13 +1,6 @@
 from .register import register_vector, register_vector_async
 
-# TODO remove
-from .. import Bit, HalfVector, SparseVector, Vector
-
 __all__ = [
     'register_vector',
     'register_vector_async',
-    'Vector',
-    'HalfVector',
-    'Bit',
-    'SparseVector'
 ]

pgvector/psycopg2/__init__.py

@@ -1,10 +1,5 @@
 from .register import register_vector
 
-# TODO remove
-from .. import HalfVector, SparseVector
-
 __all__ = [
-    'register_vector',
-    'HalfVector',
-    'SparseVector'
+    'register_vector'
 ]

pgvector/sqlalchemy/__init__.py

@@ -5,17 +5,12 @@
 from .vector import VECTOR
 from .vector import VECTOR as Vector
 
-# TODO remove
-from .. import HalfVector, SparseVector
-
 __all__ = [
     'Vector',
     'VECTOR',
     'HALFVEC',
     'BIT',
     'SPARSEVEC',
-    'HalfVector',
-    'SparseVector',
     'avg',
     'sum'
 ]