Matplotlib backend automatically set as 'Agg' when Catalyst is imported.

🐛 Bug Report

https://github.com/catalyst-team/catalyst/blob/93774200bc65f5c5ee7e2ad9935fee901c277be0/catalyst/contrib/utils/visualization.py#L6

matplotlib.use("Agg") set the backend of catalyst to agg, which requires user to click to close the visualization window. I think it would be better if it is kept in original matplotlib backend.

please make a test before allowing the pull request.

How To Reproduce

Steps to reproduce the behavior:

1. run first script, in PyCharm with Catalyst installed Env.
2. run second script, in PyCharm with Catalyst installed Env.
3. make comparison

Code sample

import numpy as np
import matplotlib.pyplot as plt
image = np.random.random([800,800,3]) * 255.0
plt.imshow(image)
plt.show()

from catalyst import dl, utils
from catalyst.data import ToTensor
from catalyst.contrib.datasets import MNIST

import numpy as np
import matplotlib.pyplot as plt
image = np.random.random([800,800,3]) * 255.0
plt.imshow(image)
plt.show()

Screenshots

Expected behavior

In pycharm, it supposed that the image should pop up in the SciView window, but after importing catalyst packages, it doesn’t appear, after commenting the matplotlib.use("Agg"), unexceptional behavior disappeared.

Environment

Additional context

Checklist

• [ x ] bug description
• [ x ] steps to reproduce
• [ x ] expected behavior
• [ x ] environment
• [ x ] code sample / screenshots

FAQ

Please review the FAQ before submitting an issue:

• [ x ] I have read the documentation and FAQ
• [ x ] I have reviewed the minimal examples section
• I have checked the changelog for main framework updates
• [ x ] I have read the contribution guide
• I have joined Catalyst slack (#__questions channel) for issue discussion

[penguinflys]

focusing only on plotting, I think it is right. An PR is already made, but not sure if it is hotfix or not.

[penguinflys]

As MR is already done, This issue is closed now