Remove the mocking of LayerNormalization in tests

As discussed in the review of #132, we're patching `tf_keras.layers.LayerNormalization` with itself in some of our tests, e.g. in gematria/granite/python/gnn_model_base_test.py. We should investigate whether this mocking is still needed, and if not, remove it.