Disable Trackio trace logging by default

Author: abidlabs

areal/api/cli_args.py

@@ -2435,7 +2435,7 @@ class TrackioConfig:
     space_id: str | None = None
     """HF Space ID for remote dashboard deployment (e.g. "user/my-space").
     When set, metrics are also pushed to the specified Hugging Face Space."""
-    max_rollout_traces_per_step: int = 32
+    max_rollout_traces_per_step: int = 0
     """Maximum rollout/eval trajectories to log as Trackio traces per step.
     Set to 0 or a negative value to disable trace logging."""
 

docs/en/cli_reference.md

@@ -844,7 +844,7 @@ See: https://github.com/gradio-app/trackio
 | `project`                     | string \| None | `None`       | -           |
 | `name`                        | string \| None | `None`       | -           |
 | `space_id`                    | string \| None | `None`       | -           |
-| `max_rollout_traces_per_step` | integer        | `32`         | -           |
+| `max_rollout_traces_per_step` | integer        | `0`          | -           |
 
 (section-wand-b)=
 

docs/zh/cli_reference.md

@@ -842,7 +842,7 @@ See: https://github.com/gradio-app/trackio
 | `project`                     | string \| None | `None`       | -           |
 | `name`                        | string \| None | `None`       | -           |
 | `space_id`                    | string \| None | `None`       | -           |
-| `max_rollout_traces_per_step` | integer        | `32`         | -           |
+| `max_rollout_traces_per_step` | integer        | `0`          | -           |
 
 (section-wand-b)=
 

tests/test_trackio_backend.py

@@ -25,7 +25,7 @@ def test_default_optional_fields_are_none(self):
         assert config.project is None
         assert config.name is None
         assert config.space_id is None
-        assert config.max_rollout_traces_per_step == 32
+        assert config.max_rollout_traces_per_step == 0
 
     def test_custom_values(self):
         """TrackioConfig should accept custom values."""
@@ -118,7 +118,9 @@ def test_trackio_init_called_when_enabled(
 
         from areal.utils.stats_logger import StatsLogger
 
-        config = _make_test_config(TrackioConfig(mode="online"))
+        config = _make_test_config(
+            TrackioConfig(mode="online", max_rollout_traces_per_step=32)
+        )
         logger = StatsLogger(config, _make_ft_spec())
         mock_trackio.init.assert_called_once()
         assert logger._trackio_enabled is True
@@ -152,7 +154,9 @@ def test_trackio_log_called_on_commit(
 
         from areal.utils.stats_logger import StatsLogger
 
-        config = _make_test_config(TrackioConfig(mode="online"))
+        config = _make_test_config(
+            TrackioConfig(mode="online", max_rollout_traces_per_step=32)
+        )
         logger = StatsLogger(config, _make_ft_spec())
         mock_trackio.log.reset_mock()
 
@@ -172,7 +176,9 @@ def test_trackio_finish_called_on_close(
 
         from areal.utils.stats_logger import StatsLogger
 
-        config = _make_test_config(TrackioConfig(mode="online"))
+        config = _make_test_config(
+            TrackioConfig(mode="online", max_rollout_traces_per_step=32)
+        )
         logger = StatsLogger(config, _make_ft_spec())
         mock_trackio.finish.reset_mock()
 
@@ -216,7 +222,9 @@ def test_trackio_trace_logging_from_rollout_tensors(
 
         from areal.utils.stats_logger import StatsLogger
 
-        config = _make_test_config(TrackioConfig(mode="online"))
+        config = _make_test_config(
+            TrackioConfig(mode="online", max_rollout_traces_per_step=32)
+        )
         logger = StatsLogger(config, _make_ft_spec())
         mock_trackio.log.reset_mock()
 
@@ -324,7 +332,9 @@ def test_trackio_trace_logging_skips_none_trajectories(
 
         from areal.utils.stats_logger import StatsLogger
 
-        config = _make_test_config(TrackioConfig(mode="online"))
+        config = _make_test_config(
+            TrackioConfig(mode="online", max_rollout_traces_per_step=32)
+        )
         logger = StatsLogger(config, _make_ft_spec())
         mock_trackio.log.reset_mock()
 
@@ -361,7 +371,9 @@ def test_trackio_trace_logging_reconstructs_multiturn_tool_messages(
 
         from areal.utils.stats_logger import StatsLogger
 
-        config = _make_test_config(TrackioConfig(mode="online"))
+        config = _make_test_config(
+            TrackioConfig(mode="online", max_rollout_traces_per_step=32)
+        )
         logger = StatsLogger(config, _make_ft_spec())
         mock_trackio.log.reset_mock()
 
@@ -401,7 +413,9 @@ def test_trackio_trace_logging_prefers_structured_messages(
 
         from areal.utils.stats_logger import StatsLogger
 
-        config = _make_test_config(TrackioConfig(mode="online"))
+        config = _make_test_config(
+            TrackioConfig(mode="online", max_rollout_traces_per_step=32)
+        )
         logger = StatsLogger(config, _make_ft_spec())
         mock_trackio.log.reset_mock()