MuJoCo Web Viewer: update code to support the web viewer

PiperOrigin-RevId: 957147035
Change-Id: Ibe5d68b2654a7faac23c297d4f492929209c078d
This commit is contained in:
Matija Kecman
2026-07-31 08:06:33 -07:00
committed by Copybara-Service
parent a1e90c24b3
commit 6c78c98418
3 changed files with 12 additions and 4 deletions
@@ -93,9 +93,7 @@ def run_viewer_target(
Raises:
ValueError: If the viewer mode requested in config is unknown.
NotImplementedError: If web mode is requested.
"""
if config.viewer_mode == viewer_protocol.ViewerMode.NATIVE:
from mujoco.experimental.studio import native_viewer # pylint: disable=g-import-not-at-top
@@ -103,7 +101,9 @@ def run_viewer_target(
config, viewer_endpoint, handlers=handlers
)
elif config.viewer_mode == viewer_protocol.ViewerMode.WEB:
raise NotImplementedError('Web viewer not implemented yet')
from mujoco.experimental.studio import web_viewer # pylint: disable=g-import-not-at-top
viewer = web_viewer.WebViewer(config, viewer_endpoint, handlers=handlers)
else:
raise ValueError(f'Unknown viewer mode: {config.viewer_mode!r}')