diff --git a/models/sign_group_user.py b/models/sign_group_user.py index f6449888..702ff4ae 100755 --- a/models/sign_group_user.py +++ b/models/sign_group_user.py @@ -50,7 +50,7 @@ class SignGroupUser(Model): @classmethod async def get_all_impression( cls, group_id: Union[int, str] - ) -> Tuple[List[str], List[str], List[float]]: + ) -> Tuple[List[str], List[float], List[str]]: """ 说明: 获取该群所有用户 id 及对应 好感度 @@ -69,7 +69,7 @@ class SignGroupUser(Model): user_list.append(value[0]) group_list.append(value[1]) impression_list.append(float(value[2])) - return user_list, group_list, impression_list + return user_list, impression_list, group_list @classmethod async def _run_script(cls): diff --git a/utils/data_utils.py b/utils/data_utils.py index 243d3805..6c6458c0 100755 --- a/utils/data_utils.py +++ b/utils/data_utils.py @@ -9,8 +9,8 @@ from utils.image_utils import BuildMat async def init_rank( title: str, - all_user_id: List[int], - all_user_data: List[int], + all_user_id: List[str], + all_user_data: List[Union[int, float]], group_id: int, total_count: int = 10, ) -> BuildMat: