code: def test_raise_on_non_string_column_name():
      df = dd.from_pandas(
          pd.DataFrame({"a": range(10), 1: range(10)}), npartitions=5
      )
      with pytest.raises(
          TypeError, match="p2p requires all column names to be str"
      ), dask.config.set({"dataframe.shuffle.method": "p2p"}):
          df.shuffle("a")
  
file_path: distributed/distributed/shuffle/tests/test_graph.py
imports: from __future__ import annotations
  import asyncio
  import pytest
  import dask
  import dask.dataframe
  from dask.blockwise import Blockwise
  from dask.utils_test import hlg_layer_topological
  from distributed.utils_test import gen_cluster
label_desc: Use image node_id 5 for calling a global function with example usage: test_raise_on_non_string_column_name() without return types