docs: def _make_trace(neuron, plane):
      '''Create the trace to be plotted'''
      for neurite in iter_neurites(neuron):
          segments = list(iter_segments(neurite))
  
          segs = [(s[0][COLS.XYZ], s[1][COLS.XYZ]) for s in segments]
  
          coords = dict(x=list(chain.from_iterable((p1[0], p2[0], None) for p1, p2 in segs)),
                        y=list(chain.from_iterable((p1[1], p2[1], None) for p1, p2 in segs)),
                        z=list(chain.from_iterable((p1[2], p2[2], None) for p1, p2 in segs)))
  
          color = TREE_COLOR.get(neurite.root_node.type, 'black')
          if plane.lower() == '3d':
              plot_fun = go.Scatter3d
          else:
              plot_fun = go.Scatter
              coords = dict(x=coords[plane[0]], y=coords[plane[1]])
          yield plot_fun(
              line=dict(color=color, width=2),
              mode='lines',
              **coords
          )