gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-04 10:47:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: privatize remain_time_duration and use remain_time() (#1027)

Browse Source
This commit is contained in:
you-n-g
2025-07-07 18:52:17 +08:00
committed by GitHub
parent a0015c1511
commit f71a005d55
3 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/log/timer.py
+6 -6
View File
@@ -10,7 +10,7 @@ class RDAgentTimer:
self.started: bool = False
self.target_time: datetime | None = None
self.all_duration: timedelta | None = None
self.remain_time_duration: timedelta | None = None
self._remain_time_duration: timedelta | None = None
def reset(self, all_duration: str | timedelta) -> None:
if isinstance(all_duration, str):
@@ -39,10 +39,10 @@ class RDAgentTimer:
return None
def restart_by_remain_time(self) -> None:
if self.remain_time_duration is not None:
self.target_time = datetime.now() + self.remain_time_duration
if self._remain_time_duration is not None:
self.target_time = datetime.now() + self._remain_time_duration
self.started = True
logger.info(f"Timer restarted with remaining time: {self.remain_time_duration}")
logger.info(f"Timer restarted with remaining time: {self._remain_time_duration}")
else:
logger.warning("No remaining time to restart the timer.")
return None
@@ -62,13 +62,13 @@ class RDAgentTimer:
def update_remain_time(self) -> None:
if self.started and self.target_time is not None:
self.remain_time_duration = self.target_time - datetime.now()
self._remain_time_duration = self.target_time - datetime.now()
return None
def remain_time(self) -> timedelta | None:
if self.started:
self.update_remain_time()
return self.remain_time_duration
return self._remain_time_duration
return None