Skip to content

Commit 6aa7ba1

Browse files
committed
Fix merge conflict
Signed-off-by: Yong Tang <[email protected]>
1 parent 566fa69 commit 6aa7ba1

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tensorflow_io/core/python/ops/io_tensor.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
import tensorflow as tf
2424
from tensorflow_io.core.python.ops import core_ops
2525
from tensorflow_io.core.python.ops import data_ops
26-
from tensorflow_io.kafka.python.ops.kafka_ops import kafka_ops
2726

2827
class _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])

0 commit comments

Comments
 (0)