From f4b2f76def38a3ed0a838c176c60826777b2b5ba Mon Sep 17 00:00:00 2001
From: Matija Kecman <matijak@google.com>
Date: Tue, 30 Jun 2026 14:16:05 -0700
Subject: [PATCH] Updates NativeViewer instantiations in sample scripts to pass
 a ViewerConfig object

PiperOrigin-RevId: 940654165
Change-Id: I984cb296593e992dd0d99ca44b43c54de4c3e96d
---
 python/mujoco/experimental/studio/sample/async.py        | 4 ++--
 python/mujoco/experimental/studio/sample/async_thread.py | 4 ++--
 python/mujoco/experimental/studio/sample/implot.py       | 5 ++---
 3 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/python/mujoco/experimental/studio/sample/async.py b/python/mujoco/experimental/studio/sample/async.py
index 892abbaf..b3b60808 100644
--- a/python/mujoco/experimental/studio/sample/async.py
+++ b/python/mujoco/experimental/studio/sample/async.py
@@ -63,13 +63,13 @@ def view(
   data = mujoco.MjData(first_msg.model)
   app = studio_app.StudioApp(first_msg.model, data)
   send_rate = 60.0
-  viewer = _viewer.NativeViewer(
-      app.model,
+  config = vp.ViewerConfig(
       title=title,
       width=_WIDTH.value,
       height=_HEIGHT.value,
       gfx=_GFX.value,
   )
+  viewer = _viewer.NativeViewer(config)
 
   while viewer.is_running() and app.is_running():
 
diff --git a/python/mujoco/experimental/studio/sample/async_thread.py b/python/mujoco/experimental/studio/sample/async_thread.py
index 5f4cbfd2..afc4d72e 100644
--- a/python/mujoco/experimental/studio/sample/async_thread.py
+++ b/python/mujoco/experimental/studio/sample/async_thread.py
@@ -63,13 +63,13 @@ def view(
 
   data = mujoco.MjData(first_msg.model)
   app = studio_app.StudioApp(first_msg.model, data)
-  viewer = _viewer.NativeViewer(
-      app.model,
+  config = vp.ViewerConfig(
       title=title,
       width=_WIDTH.value,
       height=_HEIGHT.value,
       gfx=_GFX.value,
   )
+  viewer = _viewer.NativeViewer(config)
 
   while viewer.is_running() and app.is_running():
 
diff --git a/python/mujoco/experimental/studio/sample/implot.py b/python/mujoco/experimental/studio/sample/implot.py
index e5c4dc19..df0a59b2 100644
--- a/python/mujoco/experimental/studio/sample/implot.py
+++ b/python/mujoco/experimental/studio/sample/implot.py
@@ -106,14 +106,13 @@ def main(argv: list[str]) -> None:
   app = studio_app.StudioApp.from_argv(argv)
   title = os.path.basename(sys.argv[0])
 
-  # Initialize the viewer.
-  viewer = _viewer.NativeViewer(
-      app.model,
+  config = viewer_protocol.ViewerConfig(
       title=title,
       width=_WIDTH.value,
       height=_HEIGHT.value,
       gfx=_GFX.value,
   )
+  viewer = _viewer.NativeViewer(config)
 
   # Variables for the custom UI.
   centroid = [np.zeros(3) for _ in range(_N_HISTORY)]