Fix circular import when importing test_ensemble_plot

equinor · Oct 7, 2024 · 8c38309 · 8c38309
1 parent 8b3667d
commit 8c38309
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/src/ert/gui/plottery/plot_config_factory.py b/src/ert/gui/plottery/plot_config_factory.py
@@ -1,5 +1,11 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from ert.gui.plottery import PlotConfig
-from ert.gui.tools.plot.plot_api import PlotApiKeyDefinition
+
+if TYPE_CHECKING:
+    from ert.gui.tools.plot.plot_api import PlotApiKeyDefinition
 
 
 class PlotConfigFactory:

diff --git a/src/ert/gui/plottery/plot_context.py b/src/ert/gui/plottery/plot_context.py
@@ -2,12 +2,11 @@
 
 from typing import TYPE_CHECKING, ClassVar, List, Optional
 
-from ert.gui.tools.plot.plot_api import EnsembleObject
-
 if TYPE_CHECKING:
     from pandas import DataFrame
 
     from ert.gui.plottery import PlotConfig
+    from ert.gui.tools.plot.plot_api import EnsembleObject
 
 
 class PlotContext:

diff --git a/src/ert/gui/plottery/plots/ensemble.py b/src/ert/gui/plottery/plots/ensemble.py
@@ -6,7 +6,6 @@
 import pandas as pd
 
 from ert.gui.plottery.plots.history import plotHistory
-from ert.gui.tools.plot.plot_api import EnsembleObject
 from ert.summary_key_type import is_rate
 
 from .observations import plotObservations
@@ -18,6 +17,7 @@
     from matplotlib.figure import Figure
 
     from ert.gui.plottery import PlotConfig, PlotContext
+    from ert.gui.tools.plot.plot_api import EnsembleObject
 
 
 class EnsemblePlot: