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

v1.0.5; add port cli config

Browse Source
This commit is contained in:
wty-yy
2026-01-06 22:34:46 +08:00
parent 22aedec6fb
commit 9c3184a43d
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
UPDATE.md
View File

@@ -1,4 +1,7 @@
# UPDATE # UPDATE
## 20260106
### v1.0.5
1. `robogauge/scripts/server.py`加入`--port`设置端口配置选项
## 20260103 ## 20260103
### v1.0.4 ### v1.0.4
1. 加入save_additional_output参数, 启动会自动记录每个step的latent和weights 1. 加入save_additional_output参数, 启动会自动记录每个step的latent和weights

6
robogauge/scripts/server.py
View File

@@ -20,6 +20,7 @@ import uuid
from fastapi import FastAPI from fastapi import FastAPI
from pydantic import BaseModel from pydantic import BaseModel
from typing import Dict, Optional from typing import Dict, Optional
import argparse
from dataclasses import dataclass from dataclasses import dataclass
from robogauge.utils.helpers import parse_args, class_to_dict 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") uvicorn.run(app, host="127.0.0.1", port=port, log_level="error")
def main(): 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...") print("🤖 RoboGauge Evaluation Server Starting...")
ctx = multiprocessing.get_context('spawn') ctx = multiprocessing.get_context('spawn')
manager = ctx.Manager() manager = ctx.Manager()
@@ -100,7 +104,7 @@ def main():
api_p = ctx.Process( api_p = ctx.Process(
target=run_api_server, target=run_api_server,
args=(task_queue, results_store), args=(task_queue, results_store, args_cli.port),
daemon=True daemon=True
) )
api_p.start() api_p.start()

2
setup.py
View File

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
setup( setup(
name="robogauge", # 包名 name="robogauge", # 包名
version="1.0.4", # 版本号 version="1.0.5", # 版本号
author="Wu Tianyang", # 你的名字 author="Wu Tianyang", # 你的名字
author_email="993660140@qq.com", author_email="993660140@qq.com",
description="A generic robot RL model evaluation library based on MuJoCo", description="A generic robot RL model evaluation library based on MuJoCo",