Change restore timeout to 1 hour

PiperOrigin-RevId: 718105325
AI-Hypercomputer · Jan 22, 2025 · 991c49a · 991c49a
1 parent cefacfc
commit 991c49a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pathwaysutils/__init__.py b/pathwaysutils/__init__.py
@@ -60,7 +60,7 @@ def _is_persistence_enabled():
   # TODO(b/365549911): Remove when OCDBT-compatible
   if _is_persistence_enabled():
     pathways_orbax_handler.register_pathways_handlers(
-        datetime.timedelta(minutes=10)
+        datetime.timedelta(hours=1)
     )
 
   # Turn off JAX compilation cache because Pathways handles its own compilation