Fix import to pass CI

Author: adrhill

src/innvestigate/__init__.py

@@ -1,9 +1,9 @@
 from innvestigate import analyzer  # noqa
 from innvestigate.analyzer import create_analyzer  # noqa
 from innvestigate.analyzer.base import NotAnalyzeableModelException  # noqa
-from innvestigate.backend.graph import (
-    model_wo_softmax,
+from innvestigate.backend.graph import (  # noqa
     model_wo_output_activation,
-)  # noqa
+    model_wo_softmax,
+)
 
 __version__ = "2.0.1"