Issues with Python3 Version about wgan-gp HOT 3 CLOSED

def flush():
	prints = []
	for name, vals in _since_last_flush.items():
		prints.append("{}\t{}".format(name, np.mean(vals.values())))
		_since_beginning[name].update(vals)
		x_vals = np.sort(_since_beginning[name].keys())
		y_vals = [_since_beginning[name][x] for x in x_vals]
		plt.clf()
		plt.plot(x_vals, y_vals)
		plt.xlabel('iteration')
		plt.ylabel(name)
		plt.savefig(name.replace(' ', '_')+'.jpg')

x_vals = np.sort(_since_beginning[name].keys())
y_vals = [_since_beginning[name][x] for x in x_vals]

line 845,in <module>
    lib.plot.flush()
  File "/Users/Jay/Documents/Andromeda/BitBucket_Networks/newnetworks/tflib/plot.py", line 28, in flush
    x_vals = np.sort(_since_beginning[name].keys())
  File "//anaconda/lib/python3.5/site-packages/numpy/core/fromnumeric.py", line 822, in sort
    a.sort(axis=axis, kind=kind, order=order)
numpy.core._internal.AxisError: axis-1 is out of bounds for array of dimension 0

x_vals = np.sort(_since_beginning[name].keys())

x_vals = np.sort(list(_since_beginning[name].keys()))