The current test_mnist_rotate_final_layer_invariance function doesn't actually enter the bulk of the rotate_final_layer_invariance function, so nothing is actually tested.
If you instead have "node_layers": ["layers.0", "layers.1", "layers.2"], the test fails.
The current
test_mnist_rotate_final_layer_invariance
function doesn't actually enter the bulk of therotate_final_layer_invariance
function, so nothing is actually tested.If you instead have
"node_layers": ["layers.0", "layers.1", "layers.2"]
, the test fails.Investigate why and fix this.