Missing train attribute in WandBCB

[mnslarcher]

In notebook 21, the WandBCB assumes the presence of a train attribute in _log, but this is not present.
A possible fix would be to replace it with d["train"] == "train".

If what I said is correct and you agree with the solution, I can do a PR to fix the bug.