Skip to content

Commit 47b7e20

Browse files
one validation check and a comment (#1816)
1 parent f323d22 commit 47b7e20

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/snowflake/connector/constants.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,10 @@ class FieldType(NamedTuple):
4444

4545

4646
def vector_pa_type(metadata: ResultMetadataV2) -> DataType:
47-
"""Generate the arrow type represented by the given vector column metadata."""
47+
"""Generate the Arrow type represented by the given vector column metadata.
48+
49+
Vectors are represented as Arrow fixed-size lists.
50+
"""
4851

4952
if metadata.fields is None:
5053
raise ValueError(
@@ -60,6 +63,10 @@ def vector_pa_type(metadata: ResultMetadataV2) -> DataType:
6063
raise ValueError(
6164
"Invalid result metadata for vector type: expected a dimension"
6265
)
66+
elif metadata.vector_dimension <= 0:
67+
raise ValueError(
68+
"Invalid result metadata for vector type: expected a positive dimension"
69+
)
6370

6471
if field_type == "FIXED":
6572
return pa.list_(pa.int32(), metadata.vector_dimension)

0 commit comments

Comments
 (0)