diff options
| -rw-r--r-- | tensorflow/contrib/boosted_trees/BUILD | 33 | ||||
| -rw-r--r-- | tensorflow/contrib/timeseries/BUILD | 12 | ||||
| -rw-r--r-- | tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py | 2 | ||||
| -rw-r--r-- | tensorflow/tools/pip_package/BUILD | 9 |
4 files changed, 45 insertions, 11 deletions
diff --git a/tensorflow/contrib/boosted_trees/BUILD b/tensorflow/contrib/boosted_trees/BUILD index 4f21744836..5f726cc18e 100644 --- a/tensorflow/contrib/boosted_trees/BUILD +++ b/tensorflow/contrib/boosted_trees/BUILD @@ -72,7 +72,10 @@ py_test( size = "small", srcs = ["python/kernel_tests/ensemble_optimizer_ops_test.py"], srcs_version = "PY2AND3", - tags = ["nomac"], # b/63258195 + tags = [ + "no_pip", # b/63386583 + "nomac", # b/63258195 + ], deps = [ ":ensemble_optimizer_ops_py", ":model_ops_py", @@ -91,7 +94,10 @@ py_test( size = "small", srcs = ["python/kernel_tests/model_ops_test.py"], srcs_version = "PY2AND3", - tags = ["nomac"], # b/63258195 + tags = [ + "no_pip", # b/63386583 + "nomac", # b/63258195 + ], deps = [ ":ensemble_optimizer_ops_py", ":model_ops_py", @@ -112,7 +118,10 @@ py_test( size = "small", srcs = ["python/kernel_tests/prediction_ops_test.py"], srcs_version = "PY2AND3", - tags = ["nomac"], # b/63258195 + tags = [ + "no_pip", # b/63386583 + "nomac", # b/63258195 + ], deps = [ ":model_ops_py", ":prediction_ops_py", @@ -131,7 +140,10 @@ py_test( size = "small", srcs = ["python/kernel_tests/quantile_ops_test.py"], srcs_version = "PY2AND3", - tags = ["nomac"], # b/63258195 + tags = [ + "no_pip", # b/63386583 + "nomac", # b/63258195 + ], deps = [ ":quantile_ops_py", "//tensorflow/contrib/boosted_trees/proto:quantiles_proto_py", @@ -148,6 +160,9 @@ py_test( size = "small", srcs = ["python/kernel_tests/split_handler_ops_test.py"], srcs_version = "PY2AND3", + tags = [ + "no_pip", # b/63386583 + ], deps = [ ":split_handler_ops_py", "//tensorflow/contrib/boosted_trees/proto:learner_proto_py", @@ -163,7 +178,10 @@ py_test( size = "small", srcs = ["python/kernel_tests/stats_accumulator_ops_test.py"], srcs_version = "PY2AND3", - tags = ["nomac"], # b/63258195 + tags = [ + "no_pip", # b/63386583 + "nomac", # b/63258195 + ], deps = [ ":stats_accumulator_ops_py", "//tensorflow/python:framework_test_lib", @@ -179,7 +197,10 @@ py_test( size = "small", srcs = ["python/kernel_tests/training_ops_test.py"], srcs_version = "PY2AND3", - tags = ["nomac"], # b/63258195 + tags = [ + "no_pip", # b/63386583 + "nomac", # b/63258195 + ], deps = [ ":model_ops_py", ":training_ops_py", diff --git a/tensorflow/contrib/timeseries/BUILD b/tensorflow/contrib/timeseries/BUILD index a64624bb0f..40aadb02a5 100644 --- a/tensorflow/contrib/timeseries/BUILD +++ b/tensorflow/contrib/timeseries/BUILD @@ -22,6 +22,18 @@ py_library( ], ) +py_library( + name = "timeseries_pip", + deps = [ + ":timeseries", + "//tensorflow/contrib/timeseries/examples:known_anomaly", + "//tensorflow/contrib/timeseries/examples:multivariate", + "//tensorflow/contrib/timeseries/examples:predict", + "//tensorflow/contrib/timeseries/python/timeseries:test_utils", + "//tensorflow/contrib/timeseries/python/timeseries/state_space_models:test_utils", + ], +) + filegroup( name = "all_files", srcs = glob( diff --git a/tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py b/tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py index 3b1a01ed44..f5f86ae1ad 100644 --- a/tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py +++ b/tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py @@ -66,7 +66,7 @@ class ARModelTest(test.TestCase): noise_stddev / 3. * np.random.randn(num_samples, 1)) # Add some anomalies to data1 if anomaly_prob > 0.: - num_anomalies = anomaly_prob * num_samples + num_anomalies = int(anomaly_prob * num_samples) anomaly_values = (anomaly_stddev_scale * noise_stddev / 4 * np.random.randn(num_anomalies)) indices = np.random.randint(0, num_samples, num_anomalies) diff --git a/tensorflow/tools/pip_package/BUILD b/tensorflow/tools/pip_package/BUILD index 0bd7cda1c5..78a652ccae 100644 --- a/tensorflow/tools/pip_package/BUILD +++ b/tensorflow/tools/pip_package/BUILD @@ -141,18 +141,19 @@ sh_binary( "//tensorflow/contrib/labeled_tensor:labeled_tensor_pip", "//tensorflow/contrib/ndlstm:ndlstm", "//tensorflow/contrib/nn:nn_py", + "//tensorflow/contrib/predictor:predictor_pip", "//tensorflow/contrib/session_bundle:session_bundle_pip", "//tensorflow/contrib/signal:signal_py", "//tensorflow/contrib/slim:slim", "//tensorflow/contrib/slim/python/slim/data:data_pip", "//tensorflow/contrib/slim/python/slim/nets:nets_pip", - "//tensorflow/contrib/tpu:tpu_estimator", - "//tensorflow/contrib/tpu:tpu_helper_library", - "//tensorflow/contrib/tpu:tpu_py", "//tensorflow/contrib/specs:specs", "//tensorflow/contrib/tensor_forest:init_py", "//tensorflow/contrib/tensor_forest/hybrid:hybrid_pip", - "//tensorflow/contrib/predictor:predictor_pip", + "//tensorflow/contrib/timeseries:timeseries_pip", + "//tensorflow/contrib/tpu:tpu_estimator", + "//tensorflow/contrib/tpu:tpu_helper_library", + "//tensorflow/contrib/tpu:tpu_py", "//tensorflow/examples/tutorials/mnist:package", "//tensorflow/python:distributed_framework_test_lib", "//tensorflow/python:meta_graph_testdata", |