Skip to content

Commit aba6dd3

Browse files
committed
style: fix ruff format failures
Signed-off-by: Kwangjin Ko <kwangjin.ko@sk.com>
1 parent 994462f commit aba6dd3

File tree

3 files changed

+5
-9
lines changed

3 files changed

+5
-9
lines changed

evaluation/episodic_memory/locomo_delete.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -57,10 +57,7 @@ async def process_conversation(
5757
await memory.delete_session_episodes()
5858
await memory.close()
5959

60-
tasks = [
61-
process_conversation(idx, item)
62-
for idx, item in enumerate(locomo_data)
63-
]
60+
tasks = [process_conversation(idx, item) for idx, item in enumerate(locomo_data)]
6461
await asyncio.gather(*tasks)
6562

6663

evaluation/episodic_memory/locomo_ingest.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -101,10 +101,7 @@ async def process_conversation(
101101

102102
await memory.close()
103103

104-
tasks = [
105-
process_conversation(idx, item)
106-
for idx, item in enumerate(locomo_data)
107-
]
104+
tasks = [process_conversation(idx, item) for idx, item in enumerate(locomo_data)]
108105
await asyncio.gather(*tasks)
109106

110107

evaluation/episodic_memory/locomo_search.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,9 @@ async def process_question(
9494
session_id=group_id,
9595
)
9696

97-
query_response = await memory.query_memory(query=question, limit=30, expand_context=3)
97+
query_response = await memory.query_memory(
98+
query=question, limit=30, expand_context=3
99+
)
98100

99101
if query_response is None:
100102
long_term_episodes = []

0 commit comments

Comments
 (0)