code: def visualize(ss_vae, viz, test_loader):
      if viz:
          plot_conditional_samples_ssvae(ss_vae, viz)
          mnist_test_tsne_ssvae(ssvae=ss_vae, test_loader=test_loader)
  
imports: import argparse
  import torch
  import torch.nn
  from utils.custom_mlp import MLP, Exp
  from utils.mnist_cached import MNISTCached, mkdir_p, setup_data_loaders
  from utils.vae_plots import mnist_test_tsne_ssvae, plot_conditional_samples_ssvae
  from visdom import Visdom
  import pyro
  import pyro.distributions
  from pyro.contrib.examples.util import print_and_log
  from pyro.infer import SVI, JitTrace_ELBO, JitTraceEnum_ELBO, Trace_ELBO, TraceEnum_ELBO, config_enumerate
  from pyro.optim import Adam
label_desc: Use image node_id 3 for calling a global function with example usage: visualize(ss_vae, viz, test_loader) without return types