keras/src/callbacks/swap_ema_weights.py/SwapEMAWeights/on_test_end
class SwapEMAWeights:    def on_test_end(self, logs=None):
        if self._ema_weights_in_model:
            self._swap_variables()
            self._ema_weights_in_model = False