Fix redundant double rendering of group checkboxes (#426)

Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>

Author: 3 people

.changeset/floppy-results-think.md

@@ -0,0 +1,5 @@
+---
+"trackio": minor
+---
+
+feat:Fix redundant double rendering of group checkboxes

tests/unit/test_audio_writer.py

@@ -78,8 +78,8 @@ def test_write_mp3_mono_and_stereo(tmp_path: Path, channels: int) -> None:
     "dtype, generator",
     [
         (np.int32, lambda n: (np.arange(n, dtype=np.int32) % 10000) - 5000),
-        (np.uint16, lambda n: (np.arange(n, dtype=np.uint16) % 65535)),
-        (np.uint8, lambda n: (np.arange(n, dtype=np.uint8) % 255)),
+        (np.uint16, lambda n: np.arange(n, dtype=np.uint16) % 65535),
+        (np.uint8, lambda n: np.arange(n, dtype=np.uint8) % 255),
         (np.int8, lambda n: (np.arange(n, dtype=np.int8) % 127) - 63),
     ],
 )

trackio/media/audio.py

@@ -114,17 +114,17 @@ def ensure_int16_pcm(data: np.ndarray) -> np.ndarray:
 
         converters: dict[np.dtype, callable] = {
             np.dtype(np.int16): lambda a: a,
-            np.dtype(np.int32): lambda a: (
-                (a.astype(np.int32) // 65536).astype(np.int16, copy=False)
+            np.dtype(np.int32): lambda a: (a.astype(np.int32) // 65536).astype(
+                np.int16, copy=False
             ),
-            np.dtype(np.uint16): lambda a: (
-                (a.astype(np.int32) - 32768).astype(np.int16, copy=False)
+            np.dtype(np.uint16): lambda a: (a.astype(np.int32) - 32768).astype(
+                np.int16, copy=False
             ),
-            np.dtype(np.uint8): lambda a: (
-                (a.astype(np.int32) * 257 - 32768).astype(np.int16, copy=False)
+            np.dtype(np.uint8): lambda a: (a.astype(np.int32) * 257 - 32768).astype(
+                np.int16, copy=False
             ),
-            np.dtype(np.int8): lambda a: (
-                (a.astype(np.int32) * 256).astype(np.int16, copy=False)
+            np.dtype(np.int8): lambda a: (a.astype(np.int32) * 256).astype(
+                np.int16, copy=False
             ),
         }
 

trackio/table.py

@@ -61,9 +61,11 @@ def _has_media_objects(self, dataframe: DataFrame) -> bool:
             if (
                 dataframe[col]
                 .apply(
-                    lambda x: isinstance(x, list)
-                    and len(x) > 0
-                    and isinstance(x[0], TrackioMedia)
+                    lambda x: (
+                        isinstance(x, list)
+                        and len(x) > 0
+                        and isinstance(x[0], TrackioMedia)
+                    )
                 )
                 .any()
             ):

trackio/ui/fns.py

@@ -263,24 +263,6 @@ def handle_run_checkbox_change(
     return selection
 
 
-def group_checkbox_update(
-    group_runs: list[str], selection: RunSelection
-) -> ColoredCheckboxGroup:
-    color_palette = utils.get_color_palette()
-    choice_indices = {run: i for i, run in enumerate(selection.choices)}
-    colors = [
-        color_palette[choice_indices.get(run, 0) % len(color_palette)]
-        for run in group_runs
-    ]
-    subset = utils.ordered_subset(group_runs, selection.selected)
-    return ColoredCheckboxGroup(
-        choices=group_runs,
-        value=subset,
-        colors=colors,
-        label=f"Runs ({len(group_runs)})",
-    )
-
-
 def handle_group_checkbox_change(
     group_selected: list[str] | None,
     selection: RunSelection,
@@ -289,7 +271,6 @@ def handle_group_checkbox_change(
     selection.replace_group(group_runs or [], group_selected or [])
     return (
         selection,
-        group_checkbox_update(group_runs or [], selection),
         run_checkbox_update(selection),
     )
 
@@ -303,6 +284,5 @@ def handle_group_toggle(
     selection.replace_group(group_runs or [], target)
     return (
         selection,
-        group_checkbox_update(group_runs or [], selection),
         run_checkbox_update(selection),
     )

trackio/ui/main.py

@@ -1272,7 +1272,6 @@ def render_grouped_runs(project, group_key, filter_text, selection):
                             ],
                             outputs=[
                                 run_selection_state,
-                                group_cb,
                                 run_cb,
                             ],
                             show_progress="hidden",
@@ -1288,7 +1287,7 @@ def render_grouped_runs(project, group_key, filter_text, selection):
                                 run_selection_state,
                                 gr.State(runs),
                             ],
-                            outputs=[run_selection_state, group_cb, run_cb],
+                            outputs=[run_selection_state, run_cb],
                             show_progress="hidden",
                             api_visibility="private",
                             queue=False,