sicara / tf-explain

Interpretability Methods for tf.keras models with Tensorflow 2.x
https://tf-explain.readthedocs.io
MIT License
1.02k stars 111 forks source link

Bug fix: Switched dimensions #117

Closed jpsimen closed 4 years ago

jpsimen commented 4 years ago

Working with non-square images, I found a bug in two places that switched the dimensions of the height and width axes. I changed this to the correct shape dimensions.

[x] Run tests using tox [x] Format using black

RaphaelMeudec commented 4 years ago

@jpsimen Thanks for fixing this! Could you have a quick look over the lint error? Thanks!

jpsimen commented 4 years ago

The lint error was a false positive (according to this Pylint issue). Therefore, I added # pylint: disable=unsubscriptable-object to the two lines causing the error.

RaphaelMeudec commented 4 years ago

@jpsimen Thanks for your contribution!