import tensorflow as tf

    pi_loaded.append(load_pi_ckpt(pi_ckpt_path, agent))
  return pi_loaded


def create_default_writer_and_save_dir(root_dir):
  """Creates default directories."""
  base_dir = osp.expanduser(root_dir)
  if not tf.io.gfile.exists(base_dir):
    tf.io.gfile.makedirs(base_dir)
  tag = datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')
  tb_logdir = osp.join(base_dir, tag, 'tb')
  save_dir = osp.join(base_dir, tag, 'train')
  tf.io.gfile.makedirs(tb_logdir)
  tf.io.gfile.makedirs(save_dir)
