From eb7bc5a7328bacbc8ba207de6d53fca7fc3c8692 Mon Sep 17 00:00:00 2001
From: Nat Tabris <nat@coiled.io>
Date: Tue, 27 Jun 2023 16:37:00 -0600
Subject: [PATCH] [DNM] Run on staging to collect spans + code

---
 .github/workflows/tests.yml | 3 ++-
 tests/conftest.py           | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 9b76cdbff5..fa5b5e1322 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -96,7 +96,8 @@ jobs:
       - name: Run Coiled Runtime Tests
         id: test
         env:
-          DASK_COILED__TOKEN: ${{ secrets.COILED_BENCHMARK_BOT_TOKEN }}
+          DASK_COILED__TOKEN: ${{ secrets.COILED_STAGING_BENCHMARK_BOT_TOKEN }}
+          DASK_COILED__SERVER: https://staging.coiledhq.com
           AWS_ACCESS_KEY_ID: ${{ secrets.RUNTIME_CI_BOT_AWS_ACCESS_KEY_ID }}
           AWS_SECRET_ACCESS_KEY: ${{ secrets.RUNTIME_CI_BOT_AWS_SECRET_ACCESS_KEY }}
           SNOWFLAKE_USER: ${{ secrets.SNOWFLAKE_USER }}
diff --git a/tests/conftest.py b/tests/conftest.py
index 426b8dea3b..15d16c0a54 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -68,7 +68,7 @@ def pytest_collection_modifyitems(config, items):
 
 dask.config.set(
     {
-        "coiled.account": "dask-benchmarks",
+        "coiled.account": "dask-engineering",
         "distributed.admin.system-monitor.gil.enabled": True,
     }
 )