This problem might only be visible on JUWELS, when using Jupyter Notebooks. Althou setting the python path, import statements are not working. After a discussion with Niklas, it might be related to the dynamical allocation of the phython path while execution. Moreover, he reviewed the code and pointed out some issues, potential problems.
Clara, would you support me here?