File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
tensorflow_io/core/python/ops Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change 2323import tensorflow as tf
2424from tensorflow_io .core .python .ops import core_ops
2525from tensorflow_io .core .python .ops import data_ops
26- from tensorflow_io .kafka .python .ops .kafka_ops import kafka_ops
2726
2827class _IOBaseTensor (object ):
2928 """_IOBaseTensor"""
@@ -433,11 +432,11 @@ def __init__(self,
433432 func_data = {"subscription" : subscription , "metadata" : metadata }
434433 def func_init (data ):
435434 """func_init"""
436- resource , _ , _ = kafka_ops .kafka_iterable_init (
435+ resource , _ , _ = core_ops .kafka_iterable_init (
437436 data ["subscription" ], metadata = data ["metadata" ],
438437 container = scope , shared_name = subscription )
439438 return resource
440- func_next = kafka_ops .kafka_iterable_next
439+ func_next = core_ops .kafka_iterable_next
441440
442441 dtype = tf .string
443442 shape = tf .TensorShape ([None ])
You can’t perform that action at this time.
0 commit comments