Change restore timeout to 1 hour

Pathways-on-Cloud Team · copybara-github · commit 7df564f0d597 · 2025-01-21T16:03:15.000-08:00
PiperOrigin-RevId: 718105325
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

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ def _is_persistence_enabled():`
`60`	`60`	`# TODO(b/365549911): Remove when OCDBT-compatible`
`61`	`61`	`if _is_persistence_enabled():`
`62`	`62`	`pathways_orbax_handler.register_pathways_handlers(`
`63`		`- datetime.timedelta(minutes=10)`
	`63`	`+ datetime.timedelta(hours=1)`
`64`	`64`	`)`
`65`	`65`
`66`	`66`	`# Turn off JAX compilation cache because Pathways handles its own compilation`