Fix import to pass CI

albermax · May 5, 2023 · 37aae3e · 37aae3e
1 parent 49d6f6b
commit 37aae3e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/innvestigate/__init__.py b/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"