message: Rework on tensorflow_io/kafka/__init__.py
diff: @@ -21,7 +21,7 @@ from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  -from tensorflow.contrib.kafka.python.ops.kafka_dataset_ops import KafkaDataset
  +from tensorflow_io.kafka.python.ops.kafka_dataset_ops import KafkaDataset
  from tensorflow.python.util.all_util import remove_undocumented
  