message: Update kafka_test.py to use tensorflow_io
diff: @@ -18,7 +18,7 @@ from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  -from tensorflow.contrib.kafka.python.ops import kafka_dataset_ops
  +from tensorflow_io.kafka.python.ops import kafka_dataset_ops
  from tensorflow.python.data.ops import iterator_ops
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  