fix pylint

This commit is contained in:
geekan 2023-12-19 19:15:30 +08:00
parent b7bd846769
commit 6dfa4e2c9e
5 changed files with 12 additions and 13 deletions

View file

@ -19,7 +19,7 @@ class LongTermMemory(Memory):
def __init__(self):
self.memory_storage: MemoryStorage = MemoryStorage()
super(LongTermMemory, self).__init__()
super().__init__()
self.rc = None # RoleContext
self.msg_from_recover = False
@ -37,7 +37,7 @@ class LongTermMemory(Memory):
self.msg_from_recover = False
def add(self, message: Message):
super(LongTermMemory, self).add(message)
super().add(message)
for action in self.rc.watch:
if message.cause_by == action and not self.msg_from_recover:
# currently, only add role's watching messages to its memory_storage
@ -50,7 +50,7 @@ class LongTermMemory(Memory):
1. find the short-term memory(stm) news
2. furthermore, filter out similar messages based on ltm(long-term memory), get the final news
"""
stm_news = super(LongTermMemory, self).find_news(observed, k=k) # shot-term memory news
stm_news = super().find_news(observed, k=k) # shot-term memory news
if not self.memory_storage.is_initialized:
# memory_storage hasn't initialized, use default `find_news` to get stm_news
return stm_news
@ -64,9 +64,9 @@ class LongTermMemory(Memory):
return ltm_news[-k:]
def delete(self, message: Message):
super(LongTermMemory, self).delete(message)
super().delete(message)
# TODO delete message in memory_storage
def clear(self):
super(LongTermMemory, self).clear()
super().clear()
self.memory_storage.clean()

View file

@ -58,7 +58,7 @@ class MemoryStorage(FaissStore):
return index_fpath, storage_fpath
def persist(self):
super(MemoryStorage, self).persist()
super().persist()
logger.debug(f"Agent {self.role_id} persist memory into local")
def add(self, message: Message) -> bool:

View file

@ -54,4 +54,4 @@ class ProductManager(Role):
return self._rc.todo
async def _observe(self, ignore_memory=False) -> int:
return await super(ProductManager, self)._observe(ignore_memory=True)
return await super()._observe(ignore_memory=True)

View file

@ -178,4 +178,4 @@ class QaEngineer(Role):
async def _observe(self, ignore_memory=False) -> int:
# This role has events that trigger and execute themselves based on conditions, and cannot rely on the
# content of memory to activate.
return await super(QaEngineer, self)._observe(ignore_memory=True)
return await super()._observe(ignore_memory=True)