Updating internal state for TensorFlow custom metrics (aka using non-update_state vars in metric calculation)

Question

Versions: python 3.8.2 (I\'ve also tried on 3.6.8, but I don\'t think the python version matters here), tensorflow 2.3.0, numpy 1.18.5

I\'m training a model for a classif