From 7d420bcbb7b1d5bbdd5e668f5f3bcd18096d4933 Mon Sep 17 00:00:00 2001
From: kevin-thankyou-lin
 <33344633+kevin-thankyou-lin@users.noreply.github.com>
Date: Tue, 5 Nov 2024 21:13:40 -0600
Subject: [PATCH] Fix seg demo script (#542)

Authored by Kevin Lin
---
 robosuite/demos/demo_segmentation.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/robosuite/demos/demo_segmentation.py b/robosuite/demos/demo_segmentation.py
index c901782d79..bb53b72d34 100644
--- a/robosuite/demos/demo_segmentation.py
+++ b/robosuite/demos/demo_segmentation.py
@@ -84,6 +84,7 @@ def segmentation_to_rgb(seg_im, random_colors=False):
         camera_segmentations=segmentation_level,
         camera_heights=512,
         camera_widths=512,
+        renderer="mujoco",  # for segmentation
     )
     env.reset()