Skip to content

Github Logs in WANDB Notes Section #312

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mpragnay merged 3 commits into from
Feb 23, 2026
Merged

Github Logs in WANDB Notes Section #312

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
dde8710
Added branch name, latest commit hash, latest commit message, for bet…
mpragnay Feb 23, 2026
9a42daf
Apply suggestions from code review
mpragnay Feb 23, 2026
3050145
code cleanup
mpragnay Feb 23, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions pufferlib/pufferl.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1038,6 +1038,19 @@ class WandbLogger:
def __init__(self, args, load_id=None, resume="allow"):
import wandb

try:
git_branch = subprocess.check_output(["git", "branch", "--show-current"], text=True).strip()
git_commit = subprocess.check_output(["git", "rev-parse", "HEAD"], text=True).strip()
# Get the latest commit message (subject and body)
git_commit_message = subprocess.check_output(["git", "log", "-1", "--pretty=%B"], text=True).strip()

# Format notes for the overview section
git_notes = f"**GitHub Repo State**\n\nBranch Name: {git_branch}\n\nLatest Commit Id: {git_commit}\n\nLatest Commit Message: {git_commit_message}"
except:
git_notes = (
"Error fetching git info. Make sure you're running this in a git repository and have git installed."
)

wandb.init(
id=load_id or wandb.util.generate_id(),
project=args["wandb_project"],
Expand All @@ -1047,6 +1060,7 @@ def __init__(self, args, load_id=None, resume="allow"):
resume=resume,
config=args,
name=args.get("wandb_name"),
notes=git_notes,
tags=[args["tag"]] if args["tag"] is not None else [],
)
self.wandb = wandb
Expand Down