Skip to content

Make Trackio logging much more robust#427

Merged
abidlabs merged 27 commits into
mainfrom
robust
Feb 20, 2026
Merged

Make Trackio logging much more robust#427
abidlabs merged 27 commits into
mainfrom
robust

Conversation

@abidlabs
Copy link
Copy Markdown
Member

@abidlabs abidlabs commented Feb 18, 2026

The main changes in this PR are:

When Trackio is running locally:

  • We no longer log through the Gradio Client, but instead use sqlite_storage module directly. This saves time because we no longer need to launch a Gradio server in trackio.__init__ and prevents any data from being dropped due to (internal) network issues. You'll notice Trackio starting noticeably faster 😎
  • We include tests and documentation that assert essentially unbounded levels of logging throughput

When Trackio is running on Spaces:

  • We still use the Gradio Client, and we now have more tests since this is a different path than local
  • If we are unable to log directly to the Gradio Space for whatever reason, the data is persisted locally in the SQLite database. Then on the next successful send (which could be even if the python script is re-run as long as the project name and space id are the same), the local db is flushed and the data is sent to the space.

Because of the significant changes above and some additional refactoring to remove dead code, THIS PR SHOULD BE THOROUGHLY TESTED

abidlabs and others added 8 commits February 17, 2026 15:06
Remove group_cb from handler outputs since run_selection_state.change
already triggers the render that recreates it. This eliminates a race
between the direct update and the render-triggered recreation.

Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
@gradio-pr-bot
Copy link
Copy Markdown
Contributor

gradio-pr-bot commented Feb 18, 2026

🦄 change detected

This Pull Request includes changes to the following packages.

Package Version
trackio minor

  • Make Trackio logging much more robust

✅ Changeset approved by @abidlabs

  • Maintainers can remove approval by unchecking this checkbox.

Something isn't right?

  • Maintainers can change the version label to modify the version bump.
  • If the bot has failed to detect any changes, or if this pull request needs to update multiple packages to different versions or requires a more comprehensive changelog entry, maintainers can update the changelog file directly.

@gradio-pr-bot
Copy link
Copy Markdown
Contributor

gradio-pr-bot commented Feb 18, 2026

🪼 branch checks and previews

Name Status URL
🦄 Changes detected! Details

@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@abidlabs abidlabs marked this pull request as ready for review February 19, 2026 19:21
@abidlabs abidlabs requested a review from Saba9 February 19, 2026 19:21
Comment thread trackio/run.py
auto_log_gpu: bool = False,
gpu_log_interval: float = 10.0,
):
"""
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why remove the triple-quote docstring here?

Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Accident, will ervert

@abidlabs abidlabs enabled auto-merge (squash) February 20, 2026 20:06
@abidlabs abidlabs merged commit 5aeb9ed into main Feb 20, 2026
7 of 8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants