minor

modules/util/setup.py

@@ -3,22 +3,18 @@ from __future__ import division, print_function, unicode_literals
 
 # Common imports
 import numpy as np
-import os
-import glob
-import imageio
-import h5py
 
-from scipy import misc
 from pathlib import Path
 
+
 # to make this notebook's output stable across runs
 def reset_graph(seed=42):
     tf.compat.v1.reset_default_graph()
     tf.compat.v1.set_random_seed(seed)
     np.random.seed(seed)
 
+
 # To plot pretty figures
-import matplotlib
 import matplotlib.pyplot as plt
 plt.rcParams['axes.labelsize'] = 14
 plt.rcParams['xtick.labelsize'] = 12
@@ -44,6 +40,4 @@ root_cachedir = home_dir+'/cache'
 cachepath = "{}/run-{}".format(root_cachedir, now)
 cachepath = cachepath+'.pkl'
 
-from importlib import reload
-
 gpu_devices = {0: '/device:GPU:0', 1: '/device:GPU:1', 2: '/device:GPU:2'}