either hack it into the `FlaggingCallback` or add a quick batch-style script we can use when logging the test/val sets
either hack it into the
FlaggingCallbackor add a quick batch-style script we can use when logging the test/val sets