From 86cf5062a7264e8cc41d7fb3493b51053b58d245 Mon Sep 17 00:00:00 2001 From: Alexey Kudinkin Date: Fri, 22 Nov 2024 16:56:10 -0800 Subject: [PATCH] Rebased PA overflow tests onto standalone Ray cluster; Tidying up Signed-off-by: Alexey Kudinkin --- python/ray/data/tests/test_arrow_block.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/ray/data/tests/test_arrow_block.py b/python/ray/data/tests/test_arrow_block.py index 22fe9f7ef7a0d..00ed13363f20e 100644 --- a/python/ray/data/tests/test_arrow_block.py +++ b/python/ray/data/tests/test_arrow_block.py @@ -91,7 +91,7 @@ def binary_dataset_single_file_gt_2gb(): ], ) def test_single_row_gt_2gb( - ray_start_regular_shared, + ray_start_regular, restore_data_context, binary_dataset_single_file_gt_2gb, col_name, @@ -124,7 +124,7 @@ def _id(row): ], ) def test_arrow_batch_gt_2gb( - ray_start_regular_shared, + ray_start_regular, parquet_dataset_single_column_gt_2gb, restore_data_context, op, @@ -158,10 +158,10 @@ def _id(x): print( f">>> Batch:\n" f"------\n" - "Column: 'id'" + "Column: 'id'\n" f"Values: {batch['id']}\n" f"------\n" - "Column: 'bin'" + "Column: 'bin'\n" f"Total: {total_binary_column_size / GiB} GiB\n" f"Values: {[str(v)[:3] + ' x ' + str(len(v)) for v in batch['bin']]}\n" )