My Pycharm linter complains about Flatten not being found in nn (it is actually there and the code runs smoothly but the linter cannot find it). I know this is trivial, but how can I solve it?