cyy/RoboGauge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v0.1.16; Almost finish StressPipeline

Browse Source
This commit is contained in:
wty-yy
2025-12-26 23:51:43 +08:00
parent 65490fe591
commit dad65a4e27
11 changed files with 125 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
robogauge/tasks/pipeline/level_pipeline.py
View File

@@ -15,10 +15,11 @@ from robogauge.utils.logger import Logger
level_logger = Logger() # LevelPipeline logger
class LevelPipeline:
def __init__(self, args):
def __init__(self, args, console_output: bool = True):
self.args = args
self.seeds = args.seeds
level_logger.create(args.experiment_name+'_level', args.run_name)
self.console_output = console_output
level_logger.create(args.experiment_name+'_level', args.run_name, console_output=console_output)
def run(self):
level_logger.info(f"🚀 Starting Level Searcher for '{self.args.experiment_name}'.")
@@ -52,9 +53,9 @@ class LevelPipeline:
def test_level(self, level: int):
level_logger.info(f"🔍 Testing level {level}...")
self.args.level = level
multi_pipeline = MultiPipeline(self.args)
multi_pipeline = MultiPipeline(self.args, console_output=self.console_output)
aggregated_results = multi_pipeline.run()
success_mean = float(aggregated_results['success']['mean'].split(' ')[0])
success_mean = float(aggregated_results['summary']['success']['mean'].split(' ')[0])
all_success = success_mean >= 0.8
if all_success:
level_logger.info(f"✅ Level {level} passed all tests.")