one validation check and a comment (#1816)

snowflakedb · Dec 1, 2023 · 47b7e20 · 47b7e20
1 parent f323d22
commit 47b7e20
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/snowflake/connector/constants.py b/src/snowflake/connector/constants.py
@@ -44,7 +44,10 @@ class FieldType(NamedTuple):
 
 
 def vector_pa_type(metadata: ResultMetadataV2) -> DataType:
-    """Generate the arrow type represented by the given vector column metadata."""
+    """Generate the Arrow type represented by the given vector column metadata.
+
+    Vectors are represented as Arrow fixed-size lists.
+    """
 
     if metadata.fields is None:
         raise ValueError(
@@ -60,6 +63,10 @@ def vector_pa_type(metadata: ResultMetadataV2) -> DataType:
         raise ValueError(
             "Invalid result metadata for vector type: expected a dimension"
         )
+    elif metadata.vector_dimension <= 0:
+        raise ValueError(
+            "Invalid result metadata for vector type: expected a positive dimension"
+        )
 
     if field_type == "FIXED":
         return pa.list_(pa.int32(), metadata.vector_dimension)