File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -44,7 +44,10 @@ class FieldType(NamedTuple):
4444
4545
4646def 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 )
You can’t perform that action at this time.
0 commit comments