TFTS: Increase test timeouts to avoid ASan failures

PiperOrigin-RevId: 161581264

3 files changed, 13 insertions, 7 deletions

diff --git a/tensorflow/contrib/timeseries/examples/BUILD b/tensorflow/contrib/timeseries/examples/BUILD
index c3a8d60484..c6ffc1ac00 100644
--- a/tensorflow/contrib/timeseries/examples/BUILD
+++ b/tensorflow/contrib/timeseries/examples/BUILD
@@ -25,6 +25,7 @@ py_binary(
 
 py_test(
     name = "predict_test",
+    timeout = "long",  # Moderate but for asan
     srcs = [
         "predict_test.py",
     ],
@@ -58,6 +59,7 @@ py_binary(
 
 py_test(
     name = "known_anomaly_test",
+    timeout = "long",  # Moderate but for asan
     srcs = [
         "known_anomaly_test.py",
     ],
@@ -89,6 +91,7 @@ py_binary(
 
 py_test(
     name = "multivariate_test",
+    timeout = "long",  # Moderate but for asan
     srcs = [
         "multivariate_test.py",
     ],
diff --git a/tensorflow/contrib/timeseries/python/timeseries/BUILD b/tensorflow/contrib/timeseries/python/timeseries/BUILD
index eafaee8ac4..130254f40f 100644
--- a/tensorflow/contrib/timeseries/python/timeseries/BUILD
+++ b/tensorflow/contrib/timeseries/python/timeseries/BUILD
@@ -264,6 +264,7 @@ py_library(
 
 py_test(
     name = "ar_model_test",
+    timeout = "long",  # Moderate but for asan
     srcs = [
         "ar_model_test.py",
     ],
diff --git a/tensorflow/contrib/timeseries/python/timeseries/state_space_models/BUILD b/tensorflow/contrib/timeseries/python/timeseries/state_space_models/BUILD
index 7c0c07972f..236797fbf7 100644
--- a/tensorflow/contrib/timeseries/python/timeseries/state_space_models/BUILD
+++ b/tensorflow/contrib/timeseries/python/timeseries/state_space_models/BUILD
@@ -37,20 +37,24 @@ py_library(
     ],
 )
 
-tf_py_test(
+py_test(
     name = "state_space_model_test",
+    timeout = "long",  # Moderate but for asan
     srcs = [
         "state_space_model_test.py",
     ],
-    additional_deps = [
+    srcs_version = "PY2AND3",
+    tags = [
+        "no_pip_gpu",  # b/63391119
+    ],
+    deps = [
         ":kalman_filter",
         ":state_space_model",
-        "//tensorflow/contrib/timeseries/python/timeseries:input_pipeline",
         "//tensorflow/contrib/timeseries/python/timeseries:estimators",
+        "//tensorflow/contrib/timeseries/python/timeseries:input_pipeline",
         "//tensorflow/contrib/timeseries/python/timeseries:saved_model_utils",
         "//tensorflow/contrib/timeseries/python/timeseries:state_management",
         "//tensorflow/contrib/timeseries/python/timeseries:test_utils",
-        "//third_party/py/numpy",
         "//tensorflow/python:array_ops",
         "//tensorflow/python:client_testlib",
         "//tensorflow/python:constant_op",
@@ -60,9 +64,7 @@ tf_py_test(
         "//tensorflow/python:training",
         "//tensorflow/python:variable_scope",
         "//tensorflow/python:variables",
-    ],
-    tags = [
-        "no_pip_gpu",  # b/63391119
+        "//third_party/py/numpy",
     ],
 )