From e7045deea7537f4fc7ac42e65d6bdcc13bdc6295 Mon Sep 17 00:00:00 2001 From: Shravan Goswami Date: Thu, 29 Jan 2026 00:10:24 +0530 Subject: [PATCH] fix minor docs issues in callbacks --- docs/src/callbacks.md | 4 ++-- src/callbacks.jl | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/src/callbacks.md b/docs/src/callbacks.md index bbf7783e..0156f238 100644 --- a/docs/src/callbacks.md +++ b/docs/src/callbacks.md @@ -122,7 +122,7 @@ cb = mcmc_callback( name_filter=( include=["mu", "sigma"], # Only log these parameters exclude=["_internal"], # Exclude matching names - stats=true, # Include step-level statistics + stats=false, # Include step-level statistics extras=true, # Include extra diagnostics ), ) @@ -143,8 +143,8 @@ cb = mcmc_callback( stats_options=(skip=50, thin=2), name_filter=( exclude=["_internal"], + stats=false, extras=true, - hyperparams=true, ), ) diff --git a/src/callbacks.jl b/src/callbacks.jl index 393f54a6..9c43b17a 100644 --- a/src/callbacks.jl +++ b/src/callbacks.jl @@ -278,7 +278,7 @@ Create a TensorBoard logging callback. **Requires TensorBoardLogger.jl to be loa - `true` or `:default`: Use default statistics (Mean, Variance, KHist) - requires OnlineStats - An OnlineStat or tuple of OnlineStats - requires OnlineStats - `stats_options`: NamedTuple with `thin`, `skip`, `window` -- `name_filter`: NamedTuple with `include`, `exclude`, `stats`, `hyperparams` +- `name_filter`: NamedTuple with `include`, `exclude`, `stats`, `extras` # Examples ```julia