From 02d1f22f6a3803b9d62724dbcaceca456d8affc6 Mon Sep 17 00:00:00 2001
From: Adam Hendel <hendel.adam@gmail.com>
Date: Mon, 26 Aug 2024 11:08:05 -0500
Subject: [PATCH] self hosted runner

---
 .github/workflows/extension_ci.yml      | 5 ++++-
 .github/workflows/extension_upgrade.yml | 5 ++++-
 extension/tests/integration_tests.rs    | 2 +-
 3 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/extension_ci.yml b/.github/workflows/extension_ci.yml
index 053ba13..1551f0a 100644
--- a/.github/workflows/extension_ci.yml
+++ b/.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:
diff --git a/.github/workflows/extension_upgrade.yml b/.github/workflows/extension_upgrade.yml
index 53107d8..10f80d7 100644
--- a/.github/workflows/extension_upgrade.yml
+++ b/.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
diff --git a/extension/tests/integration_tests.rs b/extension/tests/integration_tests.rs
index 3312937..bc7be2a 100644
--- a/extension/tests/integration_tests.rs
+++ b/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");