Remove optional selected_subject parameter from log_subjects method

visualization/logger.py

@@ -27,15 +27,14 @@ class SimulationLogger:
             ])
         ), timeless=True)
 
-    def log_subjects(self, subjects: List[Dict[str, Any]], selected_subject: Optional[int] = None) -> None:
+    def log_subjects(self, subjects: List[Dict[str, Any]]) -> None:
         if not subjects:
             return
 
         for idx, subject in enumerate(subjects):
             try:
                 vertices, faces = create_subject_mesh(subject)
-                subject_color = [0.8, 0.2, 0.2, 1.0] if idx == selected_subject else [
-                    0.8, 0.8, 0.8, 1.0]
+                subject_color = [0.8, 0.2, 0.2, 1.0]
 
                 rr.log(
                     f"world/subject_{idx}",

visualization/visualizer.py

@@ -27,8 +27,7 @@ def visualize_simulation(file, simulation_index: Optional[int]) -> Optional[str]
 
         logger = SimulationLogger()
         logger.log_metadata(simulation['instructions'])
-        logger.log_subjects(
-            simulation['subjects'], simulation.get('selectedSubject'))
+        logger.log_subjects(simulation['subjects'])
         logger.log_camera_trajectory(simulation['cameraFrames'])
         logger.log_camera_frames(simulation['cameraFrames'])