Skip to content

Commit

Permalink
Fix test
Browse files Browse the repository at this point in the history 
Signed-off-by: Balaji Veeramani <bveeramani@berkeley.edu>
  • Loading branch information
@bveeramani
bveeramani committed Sep 25, 2024
1 parent 44d26d1 commit b607715
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions python/ray/data/tests/test_stats.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,10 @@ def gen_expected_metrics(
):
if is_map:
metrics = [
"'average_num_outputs_per_task': N",
"'average_bytes_per_output': N",
"'average_bytes_inputs_per_task': N",
"'average_bytes_outputs_per_task': N",
"'num_inputs_received': N",
"'bytes_inputs_received': N",
"'num_task_inputs_processed': N",
Expand Down Expand Up @@ -531,6 +535,10 @@ def test_dataset__repr__(ray_start_regular_shared, restore_data_context):
" base_name=ReadRange,\n"
" number=N,\n"
" extra_metrics={\n"
" average_num_outputs_per_task: N,\n"
" average_bytes_per_output: N,\n"
" average_bytes_inputs_per_task: N,\n"
" average_bytes_outputs_per_task: N,\n"
" num_inputs_received: N,\n"
" bytes_inputs_received: N,\n"
" num_task_inputs_processed: N,\n"
Expand Down Expand Up @@ -641,6 +649,10 @@ def check_stats():
" base_name=MapBatches(<lambda>),\n"
" number=N,\n"
" extra_metrics={\n"
" average_num_outputs_per_task: N,\n"
" average_bytes_per_output: N,\n"
" average_bytes_inputs_per_task: N,\n"
" average_bytes_outputs_per_task: N,\n"
" num_inputs_received: N,\n"
" bytes_inputs_received: N,\n"
" num_task_inputs_processed: N,\n"
Expand Down Expand Up @@ -706,6 +718,10 @@ def check_stats():
" base_name=ReadRange,\n"
" number=N,\n"
" extra_metrics={\n"
" average_num_outputs_per_task: N,\n"
" average_bytes_per_output: N,\n"
" average_bytes_inputs_per_task: N,\n"
" average_bytes_outputs_per_task: N,\n"
" num_inputs_received: N,\n"
" bytes_inputs_received: N,\n"
" num_task_inputs_processed: N,\n"
Expand Down

0 comments on commit b607715

Please sign in to comment.