diff --git a/spinoffs/autobnn/autobnn/BUILD b/spinoffs/autobnn/autobnn/BUILD
index 79e53539e2..75f799bcdd 100644
--- a/spinoffs/autobnn/autobnn/BUILD
+++ b/spinoffs/autobnn/autobnn/BUILD
@@ -111,6 +111,7 @@ py_library(
 
 py_test(
     name = "estimators_test",
+    timeout = "long",
     srcs = ["estimators_test.py"],
     # TODO(b/322864412): enable this test in OSS.
     tags = ["no-oss-ci"],