code: def viz(sagemaker_session):
      return visualizer.LineageTableVisualizer(sagemaker_session)
  
imports: from __future__ import absolute_import
  import pytest
  from sagemaker.lineage import visualizer
  import unittest.mock
label_desc: Use image node_id 2 for calling a global function with example usage: viz(sagemaker_session) and returns: visualizer