diff options
4 files changed, 3 insertions, 36 deletions
diff --git a/tensorflow/contrib/timeseries/examples/BUILD b/tensorflow/contrib/timeseries/examples/BUILD index c6ffc1ac00..fc18664722 100644 --- a/tensorflow/contrib/timeseries/examples/BUILD +++ b/tensorflow/contrib/timeseries/examples/BUILD @@ -33,9 +33,6 @@ py_test( "data/period_trend.csv", ], srcs_version = "PY2AND3", - tags = [ - "no_pip_gpu", # b/63391119 - ], deps = [ ":predict", ], @@ -64,9 +61,6 @@ py_test( "known_anomaly_test.py", ], srcs_version = "PY2AND3", - tags = [ - "no_pip_gpu", # b/63391119 - ], deps = [ ":known_anomaly", ], @@ -96,9 +90,6 @@ py_test( "multivariate_test.py", ], srcs_version = "PY2AND3", - tags = [ - "no_pip_gpu", # b/63391119 - ], deps = [ ":multivariate", ], diff --git a/tensorflow/contrib/timeseries/python/timeseries/BUILD b/tensorflow/contrib/timeseries/python/timeseries/BUILD index 130254f40f..e854910f85 100644 --- a/tensorflow/contrib/timeseries/python/timeseries/BUILD +++ b/tensorflow/contrib/timeseries/python/timeseries/BUILD @@ -269,9 +269,6 @@ py_test( "ar_model_test.py", ], srcs_version = "PY2AND3", - tags = [ - "no_pip_gpu", # b/63391119 - ], deps = [ ":ar_model", ":estimators", diff --git a/tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py b/tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py index f5f86ae1ad..1e1ca4e77f 100644 --- a/tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py +++ b/tensorflow/contrib/timeseries/python/timeseries/ar_model_test.py @@ -279,14 +279,14 @@ class ARModelTest(test.TestCase): variables.global_variables_initializer().run() raw_evaluation_evaled, chunked_evaluation_evaled = sess.run( [raw_evaluation, chunked_evaluation]) - self.assertAllEqual(chunked_evaluation_evaled.loss, + self.assertAllClose(chunked_evaluation_evaled.loss, raw_evaluation_evaled.loss) last_chunk_evaluation_state = [ state[-1, None] for state in chunked_evaluation_evaled.end_state] for last_chunk_state_member, raw_state_member in zip( last_chunk_evaluation_state, raw_evaluation_evaled.end_state): - self.assertAllEqual(last_chunk_state_member, raw_state_member) + self.assertAllClose(last_chunk_state_member, raw_state_member) self.assertAllEqual([[5, 7, 11]], raw_evaluation_evaled.prediction_times) for feature_name in raw_evaluation.predictions: @@ -294,7 +294,7 @@ class ARModelTest(test.TestCase): [1, 3, 1], # batch, window, num_features. The window size has 2 # cut off for the first input_window. raw_evaluation_evaled.predictions[feature_name].shape) - self.assertAllEqual( + self.assertAllClose( np.reshape(chunked_evaluation_evaled.predictions[feature_name], [-1]), np.reshape(raw_evaluation_evaled.predictions[feature_name], diff --git a/tensorflow/contrib/timeseries/python/timeseries/state_space_models/BUILD b/tensorflow/contrib/timeseries/python/timeseries/state_space_models/BUILD index 236797fbf7..57ff15e00d 100644 --- a/tensorflow/contrib/timeseries/python/timeseries/state_space_models/BUILD +++ b/tensorflow/contrib/timeseries/python/timeseries/state_space_models/BUILD @@ -44,9 +44,6 @@ py_test( "state_space_model_test.py", ], srcs_version = "PY2AND3", - tags = [ - "no_pip_gpu", # b/63391119 - ], deps = [ ":kalman_filter", ":state_space_model", @@ -101,9 +98,6 @@ tf_py_test( "//tensorflow/python:dtypes", "//tensorflow/python:math_ops", ], - tags = [ - "no_pip_gpu", # b/63391119 - ], ) py_library( @@ -138,9 +132,6 @@ tf_py_test( "//tensorflow/python:client_testlib", "//tensorflow/python:dtypes", ], - tags = [ - "no_pip_gpu", # b/63391119 - ], ) py_library( @@ -171,9 +162,6 @@ tf_py_test( "//tensorflow/python:client_testlib", "//tensorflow/python:dtypes", ], - tags = [ - "no_pip_gpu", # b/63391119 - ], ) py_library( @@ -198,9 +186,6 @@ py_test( "structural_ensemble_test.py", ], srcs_version = "PY2AND3", - tags = [ - "no_pip_gpu", # b/63391119 - ], deps = [ ":state_space_model", ":structural_ensemble", @@ -248,9 +233,6 @@ tf_py_test( "//tensorflow/python:variables", "//tensorflow/python/estimator:estimator_py", ], - tags = [ - "no_pip_gpu", # b/63391119 - ], ) py_library( @@ -281,9 +263,6 @@ tf_py_test( "//tensorflow/python:constant_op", "//tensorflow/python:dtypes", ], - tags = [ - "no_pip_gpu", # b/63391119 - ], ) py_library( |