chore(web ports): adding global variables for default foxglove/rerun web ports

This commit is contained in:
CarolinePascal
2026-07-01 11:06:55 +02:00
parent b076cfd131
commit 2c4b511b37
+7 -2
View File
@@ -88,6 +88,9 @@ from lerobot.datasets import LeRobotDataset
from lerobot.utils.constants import ACTION, DONE, OBS_STATE, REWARD, SUCCESS from lerobot.utils.constants import ACTION, DONE, OBS_STATE, REWARD, SUCCESS
from lerobot.utils.utils import init_logging from lerobot.utils.utils import init_logging
DEFAULT_FOXGLOVE_PORT = 8765
DEFAULT_RERUN_PORT = 9090
def get_feature_names(dataset: LeRobotDataset, key: str) -> list[str]: def get_feature_names(dataset: LeRobotDataset, key: str) -> list[str]:
"""Return per-dimension names for a feature from the dataset metadata. """Return per-dimension names for a feature from the dataset metadata.
@@ -175,7 +178,7 @@ def visualize_dataset(
dataset, dataset,
episode_index, episode_index,
host=host, host=host,
port=web_port if web_port is not None else 8765, port=web_port if web_port is not None else DEFAULT_FOXGLOVE_PORT,
compress_images=display_compressed_images, compress_images=display_compressed_images,
autoplay=autoplay, autoplay=autoplay,
) )
@@ -218,7 +221,9 @@ def visualize_dataset(
server_uri = rr.serve_grpc(grpc_port=grpc_port) server_uri = rr.serve_grpc(grpc_port=grpc_port)
logging.info(f"Connect to a Rerun Server: rerun rerun+http://IP:{grpc_port}/proxy") logging.info(f"Connect to a Rerun Server: rerun rerun+http://IP:{grpc_port}/proxy")
rr.serve_web_viewer( rr.serve_web_viewer(
open_browser=False, web_port=web_port if web_port is not None else 9090, connect_to=server_uri open_browser=False,
web_port=web_port if web_port is not None else DEFAULT_RERUN_PORT,
connect_to=server_uri,
) )
logging.info("Logging to Rerun") logging.info("Logging to Rerun")