Improved imports

Author: ankane

pgvector/asyncpg/__init__.py

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

pgvector/asyncpg/register.py

@@ -1,4 +1,4 @@
-from ..utils import Vector, HalfVector, SparseVector
+from .. import Vector, HalfVector, SparseVector
 
 
 async def register_vector(conn, schema='public'):

pgvector/django/__init__.py

@@ -5,7 +5,9 @@
 from .indexes import IvfflatIndex, HnswIndex
 from .sparsevec import SparseVectorField
 from .vector import VectorField
-from ..utils import HalfVector, SparseVector
+
+# TODO remove
+from .. import HalfVector, SparseVector
 
 __all__ = [
     'VectorExtension',

pgvector/django/functions.py

@@ -1,5 +1,5 @@
 from django.db.models import FloatField, Func, Value
-from ..utils import Vector, HalfVector, SparseVector
+from .. import Vector, HalfVector, SparseVector
 
 
 class DistanceBase(Func):

pgvector/django/halfvec.py

@@ -1,6 +1,6 @@
 from django import forms
 from django.db.models import Field
-from ..utils import HalfVector
+from .. import HalfVector
 
 
 # https://docs.djangoproject.com/en/5.0/howto/custom-model-fields/

pgvector/django/sparsevec.py

@@ -1,6 +1,6 @@
 from django import forms
 from django.db.models import Field
-from ..utils import SparseVector
+from .. import SparseVector
 
 
 # https://docs.djangoproject.com/en/5.0/howto/custom-model-fields/

pgvector/django/vector.py

@@ -1,7 +1,7 @@
 from django import forms
 from django.db.models import Field
 import numpy as np
-from ..utils import Vector
+from .. import Vector
 
 
 # https://docs.djangoproject.com/en/5.0/howto/custom-model-fields/

pgvector/peewee/__init__.py

@@ -2,7 +2,9 @@
 from .halfvec import HalfVectorField
 from .sparsevec import SparseVectorField
 from .vector import VectorField
-from ..utils import HalfVector, SparseVector
+
+# TODO remove
+from .. import HalfVector, SparseVector
 
 __all__ = [
     'VectorField',

pgvector/peewee/halfvec.py

@@ -1,5 +1,5 @@
 from peewee import Expression, Field
-from ..utils import HalfVector
+from .. import HalfVector
 
 
 class HalfVectorField(Field):

pgvector/peewee/sparsevec.py

@@ -1,5 +1,5 @@
 from peewee import Expression, Field
-from ..utils import SparseVector
+from .. import SparseVector
 
 
 class SparseVectorField(Field):