code: def _get_collection(node_info):
      def _to_collection(x):
          for idx, q in enumerate(x):
              tmp = DataCollection(q)
              q.reset_size()
              x[idx] = tmp
  
      for v in node_info.values():
          _to_collection(v["in"])
          _to_collection(v["out"])
  
      return node_info
  
file_path: towhee/tests/unittests/tools/test_data_visualizer.py
imports: import unittest
  from towhee import pipe
  from towhee.tools.data_visualizer import PipeVisualizer
  from towhee.datacollection import DataCollection
  from towhee.runtime.data_queue import DataQueue, ColumnType
label_desc: Use image node_id 3 for calling a global function with example usage: _get_collection(node_info) and returns: node_info