message: Add six and numpy imports
diff: @@ -22,6 +22,11 @@ from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  +
  +import six
  +from six.moves import xrange # pylint: disable=redefined-builtin
  +import numpy as np
  +
  from tensorflow.python.eager import context
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_shape
  