v1.0.5; add port cli config

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

View File

@@ -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

View File

@@ -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()

View File

@@ -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",