add param

snowflakedb · Jan 3, 2024 · e5b9be0 · e5b9be0
1 parent cd3bac5
commit e5b9be0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/driver_test.go b/driver_test.go
@@ -1286,6 +1286,8 @@ func testVectorInt(t *testing.T, json bool) {
 		},
 	}
 	runDBTest(t, func(dbt *DBTest) {
+		dbt.mustExec("alter session set enable_vector_data_type='Enable'")
+		defer dbt.mustExec("alter session unset enable_vector_data_type")
 		if json {
 			dbt.mustExec(forceJSON)
 		}
@@ -1332,9 +1334,12 @@ func testVectorFloat(t *testing.T, json bool) {
 		},
 	}
 	runDBTest(t, func(dbt *DBTest) {
+		dbt.mustExec("alter session set enable_vector_data_type='Enable'")
+		defer dbt.mustExec("alter session unset enable_vector_data_type")
 		if json {
 			dbt.mustExec(forceJSON)
 		}
+
 		for _, tt := range tests {
 			t.Run(tt.msg, func(t *testing.T) {
 				rowsFloat := dbt.mustQuery(tt.query)