set runner ray

Eventual-Inc · Dec 10, 2024 · 4e2f4fb · 4e2f4fb
1 parent 4198528
commit 4e2f4fb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/benchmarking/ooms/big_task_heap_usage.py b/benchmarking/ooms/big_task_heap_usage.py
@@ -47,6 +47,8 @@ def generator(
     parser.add_argument("--deflation-factor", type=int, default=100)
     args = parser.parse_args()
 
+    daft.context.set_runner_ray()
+
     df = read_generator(
         generator(args.num_partitions, args.num_rows_per_partition),
         schema=daft.Schema._from_field_name_and_types([("foo", daft.DataType.binary())]),