Skip to content

Commit 75a5d50

Browse files
ankaneankane
committed
Improved code [skip ci]
1 parent 107b81e commit 75a5d50

4 files changed

Lines changed: 21 additions & 8 deletions

File tree

‎pgvector/asyncpg/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33

44
__all__ = [
55
'register_vector',
6+
'Vector',
67
'HalfVector',
78
'SparseVector'
89
]

‎pgvector/sqlalchemy/__init__.py‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,12 @@
88

99
__all__ = [
1010
'Vector',
11-
'VECTOR'
11+
'VECTOR',
1212
'HALFVEC',
1313
'BIT',
1414
'SPARSEVEC',
1515
'HalfVector',
1616
'SparseVector',
17-
# TODO add avg and sum in 0.4.0
18-
# 'avg',
19-
# 'sum'
17+
'avg',
18+
'sum'
2019
]

‎pgvector/sqlalchemy/functions.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,3 +6,9 @@
66
class avg(ReturnTypeFromArgs):
77
inherit_cache = True
88
package = 'pgvector'
9+
10+
11+
__all__ = [
12+
'avg',
13+
'sum'
14+
]

‎pgvector/utils/__init__.py‎

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,11 @@
1-
from .bit import *
2-
from .halfvec import *
3-
from .sparsevec import *
4-
from .vector import *
1+
from .bit import Bit
2+
from .halfvec import HalfVector
3+
from .sparsevec import SparseVector
4+
from .vector import Vector
5+
6+
__all__ = [
7+
'Vector',
8+
'HalfVector',
9+
'Bit',
10+
'SparseVector'
11+
]

0 commit comments

Comments
 (0)