diff --git a/tensorflow_addons/custom_ops/text/BUILD b/tensorflow_addons/custom_ops/text/BUILD index 3e5c888c01..a52e139228 100644 --- a/tensorflow_addons/custom_ops/text/BUILD +++ b/tensorflow_addons/custom_ops/text/BUILD @@ -12,11 +12,10 @@ custom_op_library( ], ) -# TODO: https://github.com/tensorflow/addons/issues/663 -# custom_op_library( -# name = "_parse_time_op.so", -# srcs = [ -# "cc/kernels/parse_time_kernel.cc", -# "cc/ops/parse_time_op.cc", -# ], -# ) +custom_op_library( + name = "_parse_time_op.so", + srcs = [ + "cc/kernels/parse_time_kernel.cc", + "cc/ops/parse_time_op.cc", + ], +) diff --git a/tensorflow_addons/text/BUILD b/tensorflow_addons/text/BUILD index 30b5e010bc..fbeafe16f7 100644 --- a/tensorflow_addons/text/BUILD +++ b/tensorflow_addons/text/BUILD @@ -2,17 +2,16 @@ licenses(["notice"]) # Apache 2.0 package(default_visibility = ["//visibility:public"]) -# TODO: https://github.com/tensorflow/addons/issues/663 py_library( name = "text", srcs = ([ "__init__.py", "crf.py", - # "parse_time_op.py", + "parse_time_op.py", "skip_gram_ops.py", ]), data = [ - # "//tensorflow_addons/custom_ops/text:_parse_time_op.so", + "//tensorflow_addons/custom_ops/text:_parse_time_op.so", "//tensorflow_addons/custom_ops/text:_skip_gram_ops.so", "//tensorflow_addons/utils", ], @@ -45,16 +44,15 @@ py_test( ], ) -# TODO: https://github.com/tensorflow/addons/issues/663 -# py_test( -# name = "parse_time_op_test", -# size = "small", -# srcs = [ -# "parse_time_op_test.py", -# ], -# main = "parse_time_op_test.py", -# srcs_version = "PY2AND3", -# deps = [ -# ":text", -# ], -# ) +py_test( + name = "parse_time_op_test", + size = "small", + srcs = [ + "parse_time_op_test.py", + ], + main = "parse_time_op_test.py", + srcs_version = "PY2AND3", + deps = [ + ":text", + ], +) diff --git a/tensorflow_addons/text/__init__.py b/tensorflow_addons/text/__init__.py index d13db57b13..d334b3db81 100644 --- a/tensorflow_addons/text/__init__.py +++ b/tensorflow_addons/text/__init__.py @@ -34,6 +34,5 @@ from tensorflow_addons.text.skip_gram_ops import skip_gram_sample from tensorflow_addons.text.skip_gram_ops import skip_gram_sample_with_text_vocab -# TODO: https://github.com/tensorflow/addons/issues/663 # Parse Time -# from tensorflow_addons.text.parse_time_op import parse_time +from tensorflow_addons.text.parse_time_op import parse_time