Remove old heartbeat flags and arguments.

mwhittaker · Orbax Authors · commit ee324009d30a · 2025-06-23T10:17:58.000-07:00
PiperOrigin-RevId: 770823735
diff --git a/checkpoint/orbax/checkpoint/_src/testing/multiprocess_test.py b/checkpoint/orbax/checkpoint/_src/testing/multiprocess_test.py
@@ -162,8 +162,6 @@ def _main(argv):
         f"--jax_task_id={i}",
         f"--jax_controller_address=localhost:{jax_port}",
         "--jax_heartbeat_timeout=3s",
-        "--jax_heartbeat_interval=1s",
-        "--jax_max_missing_heartbeats=3",
         f"--jax_distributed_shutdown_timeout={_WORKER_SHUTDOWN_TIMEOUT.value}s",
         "--vmodule=client=10,service=10",
     ]

Original file line number	Diff line number	Diff line change
`@@ -162,8 +162,6 @@ def _main(argv):`
`162`	`162`	`f"--jax_task_id={i}",`
`163`	`163`	`f"--jax_controller_address=localhost:{jax_port}",`
`164`	`164`	`"--jax_heartbeat_timeout=3s",`
`165`		`- "--jax_heartbeat_interval=1s",`
`166`		`- "--jax_max_missing_heartbeats=3",`
`167`	`165`	`f"--jax_distributed_shutdown_timeout={_WORKER_SHUTDOWN_TIMEOUT.value}s",`
`168`	`166`	`"--vmodule=client=10,service=10",`
`169`	`167`	`]`