Set the runner to ray

Eventual-Inc · Dec 5, 2024 · 2515ee0 · 2515ee0
1 parent ec7f660
commit 2515ee0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/working-dir/dedup.py b/.github/working-dir/dedup.py
@@ -96,6 +96,7 @@ def components(df: DataFrame) -> DataFrame:
 
 
 if __name__ == "__main__":
+    daft.context.set_runner_ray()
     daft.set_execution_config(enable_ray_tracing=True)
 
     df = daft.read_parquet("s3://eventual-dev-benchmarking-fixtures/redpajama-parquet/v1.0.0/sample-0.01")