diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/datasets/BUILD')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/datasets/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/contrib/learn/python/learn/datasets/BUILD b/tensorflow/contrib/learn/python/learn/datasets/BUILD index 68760e2653..3ece16170a 100644 --- a/tensorflow/contrib/learn/python/learn/datasets/BUILD +++ b/tensorflow/contrib/learn/python/learn/datasets/BUILD @@ -18,6 +18,7 @@ py_library( "base.py", "mnist.py", "produce_small_datasets.py", + "synthetic.py", "text_datasets.py", ], data = [":data_csv"], @@ -76,3 +77,15 @@ py_test( "//tensorflow/python:framework_test_lib", ], ) + +py_test( + name = "synthetic_test", + size = "small", + srcs = ["synthetic_test.py"], + srcs_version = "PY2AND3", + deps = [ + "//tensorflow:tensorflow_py", + "//tensorflow/contrib/learn", + "//tensorflow/python:framework_test_lib", + ], +) |