diff --git a/UPDATE.md b/UPDATE.md index 0dd41f2..c08fe88 100644 --- a/UPDATE.md +++ b/UPDATE.md @@ -1,4 +1,7 @@ # UPDATE +## 20260106 +### v1.0.5 +1. `robogauge/scripts/server.py`加入`--port`设置端口配置选项 ## 20260103 ### v1.0.4 1. 加入save_additional_output参数, 启动会自动记录每个step的latent和weights diff --git a/robogauge/scripts/server.py b/robogauge/scripts/server.py index 053a803..da9c28a 100644 --- a/robogauge/scripts/server.py +++ b/robogauge/scripts/server.py @@ -20,6 +20,7 @@ import uuid from fastapi import FastAPI from pydantic import BaseModel from typing import Dict, Optional +import argparse from dataclasses import dataclass from robogauge.utils.helpers import parse_args, class_to_dict @@ -92,6 +93,9 @@ def run_api_server(input_queue: multiprocessing.Queue, result_dict: dict, port=9 uvicorn.run(app, host="127.0.0.1", port=port, log_level="error") def main(): + parser = argparse.ArgumentParser() + parser.add_argument('--port', type=int, default=9973, help='API server port') + args_cli = parser.parse_args() print("🤖 RoboGauge Evaluation Server Starting...") ctx = multiprocessing.get_context('spawn') manager = ctx.Manager() @@ -100,7 +104,7 @@ def main(): api_p = ctx.Process( target=run_api_server, - args=(task_queue, results_store), + args=(task_queue, results_store, args_cli.port), daemon=True ) api_p.start() diff --git a/setup.py b/setup.py index 411bf5a..9fb93c5 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import setup, find_packages setup( name="robogauge", # 包名 - version="1.0.4", # 版本号 + version="1.0.5", # 版本号 author="Wu Tianyang", # 你的名字 author_email="993660140@qq.com", description="A generic robot RL model evaluation library based on MuJoCo",