Skip to content

Commit

Permalink
Add new env-flag
Browse files Browse the repository at this point in the history
  • Loading branch information
Raunak Bhagat committed Nov 19, 2024
1 parent 5e54261 commit dbb07d2
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions .github/workflows/run-tpch.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -82,8 +82,10 @@ jobs:
ray up .github/assets/ray.yaml -y
HEAD_NODE_IP=$(ray get-head-ip .github/assets/ray.yaml | tail -n 1)
ssh -o StrictHostKeyChecking=no -fN -L 8265:localhost:8265 -i ~/.ssh/ci-github-actions-ray-cluster-key.pem ubuntu@$HEAD_NODE_IP
export DAFT_ENABLE_RAY_TRACING=1
export DAFT_RUNNER=ray
if [[ -n "${{ inputs.skip_questions }}" ]]; then
DAFT_RUNNER=ray python -m benchmarking.tpch \
python -m benchmarking.tpch \
--scale_factor ${{ inputs.scale_factor }} \
--num_parts ${{ inputs.partition_size }} \
--parquet_file_cache /tmp/data \
Expand All @@ -93,7 +95,7 @@ jobs:
--no_pymodules \
--skip_questions="${{ inputs.skip_questions }}"
else
DAFT_RUNNER=ray python -m benchmarking.tpch \
python -m benchmarking.tpch \
--scale_factor ${{ inputs.scale_factor }} \
--num_parts ${{ inputs.partition_size }} \
--parquet_file_cache /tmp/data \
Expand All @@ -106,6 +108,7 @@ jobs:
- uses: lhotari/action-upterm@v1

- run: |
source .venv/bin/activate
ray down .github/assets/ray.yaml -y
python .github/scripts/csv_to_md.py output.csv output.md
echo "# Results" >> $GITHUB_STEP_SUMMARY
Expand Down

0 comments on commit dbb07d2

Please sign in to comment.