scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-06 08:16:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

bugfix

Browse source
This commit is contained in:
jinli.yl 2025-06-18 14:18:19 +08:00
parent 5da12da545
commit 53304e189c
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
experiencemaker/module/summarizer/step_summarizer.py
View file

@ -1,11 +1,9 @@
import asyncio
import json
import re
import uuid
import asyncio
from datetime import datetime
from pathlib import Path
from typing import List, Dict, Any, Optional, Tuple
from concurrent.futures import ThreadPoolExecutor
from pathlib import Path
from typing import List, Dict, Any, Tuple
from loguru import logger
from pydantic import Field
@ -80,8 +78,8 @@ class StepSummarizer(BaseSummarizer, PromptMixin):
all_experiences = []
# Classify trajectories based on trajectory.done
success_trajectories = [traj for traj in trajectories if traj.is_terminated]
failure_trajectories = [traj for traj in trajectories if not traj.is_terminated]
success_trajectories = [traj for traj in trajectories if traj.reward.success]
failure_trajectories = [traj for traj in trajectories if not traj.reward.success]
# Process success and failure samples concurrently
tasks = []