Change autostop to use public API. (#894)

run-house · Jun 16, 2024 · 8ece21a · 8ece21a
1 parent 64609cb
commit 8ece21a
Showing 1 changed file with 3 additions and 10 deletions.
diff --git a/runhouse/servers/autostop_servlet.py b/runhouse/servers/autostop_servlet.py
@@ -18,15 +18,8 @@ def set_autostop(self, value=None):
  autostop_lib.set_autostop(value, None, True)
 
  def update_autostop_in_sky_config(self):
- import pickle
+ from sky.skylet.autostop_lib import set_last_active_time_to_now
 
- from sky.skylet import configs as sky_configs
-
- autostop_mins = pickle.loads(
- sky_configs.get_config("autostop_config")
- ).autostop_idle_minutes
- if autostop_mins > 0 and (
- self._last_register is None or self._last_register < self._last_activity
- ):
- sky_configs.set_config("autostop_last_active_time", self._last_activity)
+ if self._last_register is None or self._last_register < self._last_activity:
+ set_last_active_time_to_now()
  self._last_register = self._last_activity