From 53304e189cc3f8c73678de7c5bada64817ec14fa Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Wed, 18 Jun 2025 14:18:19 +0800 Subject: [PATCH] bugfix --- experiencemaker/module/summarizer/step_summarizer.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/experiencemaker/module/summarizer/step_summarizer.py b/experiencemaker/module/summarizer/step_summarizer.py index 6e835b0c..59d4ec96 100644 --- a/experiencemaker/module/summarizer/step_summarizer.py +++ b/experiencemaker/module/summarizer/step_summarizer.py @@ -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 = []