self hosted runner

.github/workflows/extension_ci.yml

@@ -85,7 +85,10 @@ jobs:
   test:
     name: Run tests
     needs: dependencies
-    runs-on: ubuntu-22.04
+    runs-on:
+      - self-hosted
+      - dind
+      - xlarge-16x16
     services:
       # Label used to access the service container
       vector-serve:

.github/workflows/extension_upgrade.yml

@@ -22,7 +22,10 @@ on:
 jobs:
   test:
     name: Upgrade Test
-    runs-on: ubuntu-24.04
+    runs-on:
+      - self-hosted
+      - dind
+      - xlarge-16x16
     services:
       vector-serve:
         image: quay.io/tembo/vector-serve:latest

extension/tests/integration_tests.rs

@@ -793,7 +793,7 @@ async fn test_diskann_cosine() {
         .await
         .expect("failed to drop vectorscale extension");
 
-    let _executed = sqlx::query("CREATE EXTENSION EXISTS vectorscale;")
+    let _executed = sqlx::query("CREATE EXTENSION IF EXISTS vectorscale;")
         .execute(&conn)
         .await
         .expect("failed to create vectorscale extension");