Skip to content

Commit

Permalink
Add support for producing ray logs
Browse files Browse the repository at this point in the history
- still not able to produce daft logs (located inside of `/tmp/ray/session_*/logs/daft`)
- only ray logs are being produced
    - this should be fine for now?
  • Loading branch information
Raunak Bhagat committed Nov 23, 2024
1 parent b6d8326 commit 67645b0
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/run-cluster.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@ jobs:
ray job submit \
--working-dir ${{ inputs.working_dir }} \
--address http://localhost:8265 \
--runtime-env-json '{"DAFT_ENABLE_RAY_TRACING": "1"}' \
--runtime-env-json '{"DAFT_RUNNER": "1", "DAFT_ENABLE_RAY_TRACING": "1"}' \
-- ${{ inputs.command }}
# ssh breakpoint; remove later
Expand All @@ -90,7 +90,7 @@ jobs:
- name: Download log files from ray cluster
run: |
source .venv/bin/activate
ray rsync-down .github/assets/benchmarking_ray_config.yaml /tmp/ray/session_*/logs/daft ray-daft-logs
ray rsync-down .github/assets/benchmarking_ray_config.yaml /tmp/ray/session_*/logs ray-daft-logs
- name: Kill connection to ray cluster
run: |
PID=$(lsof -t -i:8265)
Expand Down
2 changes: 2 additions & 0 deletions .github/working-dir/simple.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
import daft

daft.context.set_runner_ray("ray://localhost:10001")

df = daft.from_pydict({"nums": [1, 2, 3]})
df = df.with_column("result", daft.col("nums").cbrt()).collect()
df.show()

0 comments on commit 67645b0

Please sign in to comment.