code: def test_distributed_sort():
      args = main.get_args(
          [
              "--total_data_size=1_000_000_000",
              "--num_mappers=4",
              "--num_reducers=4",
              "--num_mappers_per_round=2",
              "--ray_address=",
              "--skip_input",
              "--skip_output",
          ]
      )
      main.main(args)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_distributed_sort() without return types