Skip to content

Commit edf29a0

Browse files
authored
feat: Add option for using spiceai-dev-runners in run spicebench (#162)
* feat: Use spiceai-dev-runners in run spicebench * fix * fix: updates * fix: back to 1 inflight write
1 parent 5319358 commit edf29a0

2 files changed

Lines changed: 14 additions & 3 deletions

File tree

.github/actions/setup-cc/action.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ runs:
2323
elif [[ "${RUNNER_OS}" == "Linux" ]]; then
2424
if command -v apt-get >/dev/null 2>&1; then
2525
sudo apt-get update
26-
sudo apt-get install -y build-essential
26+
sudo apt-get install -y build-essential protobuf-compiler pkg-config libssl-dev
2727
elif command -v dnf >/dev/null 2>&1; then
2828
sudo dnf install -y gcc gcc-c++ make
2929
elif command -v yum >/dev/null 2>&1; then

.github/workflows/run_spicebench.yml

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ on:
3232
description: 'Data generation version identifier (the {version} portion of {prefix}/{scenario}/{version}/)'
3333
required: true
3434
type: string
35+
default: "1"
3536
etl_region:
3637
description: 'AWS region for the ETL S3 bucket'
3738
required: false
@@ -40,7 +41,7 @@ on:
4041
etl_sink:
4142
description: 'ETL sink (hive or adbc)'
4243
required: false
43-
default: 'hive'
44+
default: 'adbc'
4445
type: choice
4546
options:
4647
- hive
@@ -60,11 +61,19 @@ on:
6061
required: false
6162
default: false
6263
type: boolean
64+
runner:
65+
description: 'GitHub runner label to use for this workflow run'
66+
required: true
67+
default: ubuntu-latest
68+
type: choice
69+
options:
70+
- ubuntu-latest
71+
- spiceai-dev-runners
6372

6473
jobs:
6574
run-spicebench:
6675
name: Run spicebench
67-
runs-on: ubuntu-latest
76+
runs-on: ${{ github.event.inputs.runner || 'ubuntu-latest' }}
6877
timeout-minutes: 600
6978
concurrency:
7079
group: spicebench-run-${{ github.event.inputs.system_under_test || 'spice_cloud' }}-${{ github.event.inputs.scenario || 'tpch' }}
@@ -218,13 +227,15 @@ jobs:
218227
run: |
219228
set -euo pipefail
220229
curl -LsSf https://dbc.columnar.tech/install.sh | sh
230+
source $HOME/.local/bin/env
221231
dbc install postgresql
222232
223233
- name: Install ADBC FlightSQL driver
224234
if: ${{ !startsWith(github.event.inputs.system_under_test || 'spice_cloud', 'databricks-') }}
225235
run: |
226236
set -euo pipefail
227237
curl -LsSf https://dbc.columnar.tech/install.sh | sh
238+
source $HOME/.local/bin/env
228239
dbc install flightsql
229240
230241
- name: Run spicebench

0 commit comments

Comments
 (0)