Custom Callback Example - Tensorflow Typing CST Test