Skip to content

Commit 72dd732

Browse files
authored
Rename tfio.IOLayer to tfio.experimental.IOLayer (#660)
As IOLayer was only recetnly introduced, it makes sense to first place it under tfio.experimental. Signed-off-by: Yong Tang <[email protected]>
1 parent 243fbaf commit 72dd732

File tree

6 files changed

+6
-6
lines changed

6 files changed

+6
-6
lines changed

tensorflow_io/core/python/api/experimental/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,9 @@
1414
# ==============================================================================
1515
"""tensorflow_io.experimental"""
1616

17-
from tensorflow_io.core.python.experimental.io_tensor import IOTensor
1817
from tensorflow_io.core.python.experimental.io_dataset_ops import IODataset
18+
from tensorflow_io.core.python.experimental.io_tensor import IOTensor
19+
from tensorflow_io.core.python.experimental.io_layer import IOLayer
1920

2021
from tensorflow_io.core.python.api.experimental import ffmpeg
2122
from tensorflow_io.core.python.api.experimental import image

tensorflow_io/core/python/api/v0/__init__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
# tensorflow_io.core.python.ops is implicitly imported (along with file system)
1818
from tensorflow_io.core.python.ops.io_dataset import IODataset
1919
from tensorflow_io.core.python.ops.io_tensor import IOTensor
20-
from tensorflow_io.core.python.ops.io_layer import IOLayer
2120

2221
from tensorflow_io.core.python.api.v0 import genome
2322
from tensorflow_io.core.python.api.v0 import image

tensorflow_io/core/python/ops/io_layer.py renamed to tensorflow_io/core/python/experimental/io_layer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818
from __future__ import print_function
1919

2020
import tensorflow as tf
21-
from tensorflow_io.core.python.ops import text_io_layer_ops
22-
from tensorflow_io.core.python.ops import kafka_io_layer_ops
21+
from tensorflow_io.core.python.experimental import text_io_layer_ops
22+
from tensorflow_io.core.python.experimental import kafka_io_layer_ops
2323

2424
class IOLayer(tf.keras.layers.Layer):
2525
"""IOLayer

tests/test_io_layer_eager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def func(self):
7878
f, filename = tempfile.mkstemp()
7979
os.close(f)
8080
self._filename = filename
81-
return tfio.IOLayer.text(filename)
81+
return tfio.experimental.IOLayer.text(filename)
8282
def check(self, images, predictions):
8383
f = tf.data.TextLineDataset(self._filename)
8484
lines = [line for line in f]
@@ -91,7 +91,7 @@ class KafkaIOLayerHelper(object):
9191
def func(self):
9292
channel = "e{}e".format(time.time())
9393
self._topic = "io-layer-test-"+channel
94-
return tfio.IOLayer.kafka(self._topic)
94+
return tfio.experimental.IOLayer.kafka(self._topic)
9595
def check(self, images, predictions):
9696
f = kafka_io.KafkaDataset(topics=[self._topic], group="test", eof=True)
9797
lines = [line for line in f]

0 commit comments

Comments
 (0)