flt/ai-titration
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source 
Former-commit-id: e60e29fb767ab2b7d8152eced7a50255ce6b06ad
This commit is contained in:
flt6
2025-07-07 18:56:33 +08:00
parent e0e1c649eb
commit b696cb71be
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
utils.py
View File

@ -11,13 +11,15 @@ from typing import List, Literal, Optional
import numpy as np import numpy as np
import requests import requests
stateType = Literal["transport", "middle", "about", "colored"]
@dataclass @dataclass
class HistoryRecord: class HistoryRecord:
"""历史记录的单个条目""" """历史记录的单个条目"""
timestamp: float timestamp: float
state: Literal["transport", "middle", "about", "colored"] state: stateType
rate: float rate: float
volume: float volume: float
image: np.ndarray image: np.ndarray
@ -51,7 +53,7 @@ class History:
def add_record( def add_record(
self, self,
timestamp: float, timestamp: float,
state: str, state: stateType,
rate: float, rate: float,
volume: float, volume: float,
image: np.ndarray, image: np.ndarray,
@ -69,7 +71,7 @@ class History:
return return
base_records = self.get_recent_records(self._base_time, timestamp) base_records = self.get_recent_records(self._base_time, timestamp)
self.base = sum([rec.rate for rec in base_records]) / len(base_records) self.base = sum([rec.rate for rec in base_records]) / len(base_records)
get_endpoint_logger().info("Base rate calculated: %.2f", self.base) logging.getLogger("Endpoint").info("Base rate calculated: %.2f", self.base)
def _cleanup_old_records(self, current_time: float): def _cleanup_old_records(self, current_time: float):
"""清理过期的历史记录""" """清理过期的历史记录"""