Try add suite back to __init__

yngve-sk · Oct 24, 2024 · 35c0182 · 35c0182
1 parent a867d62
commit 35c0182
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/everest/__init__.py b/src/everest/__init__.py
@@ -16,11 +16,13 @@
 from everest.bin.utils import export_to_csv, export_with_progress
 from everest.config_keys import ConfigKeys
 from everest.export import MetaDataColumnNames, export, filter_data, validate_export
+from everest.suite import _EverestWorkflow
 
 __author__ = "Equinor ASA and TNO"
 __all__ = [
     "ConfigKeys",
     "MetaDataColumnNames",
+    "_EverestWorkflow",
     "detached",
     "docs",
     "export",

diff --git a/src/everest/suite.py b/src/everest/suite.py
@@ -7,11 +7,12 @@
 from ert.config import QueueSystem
 from ert.ensemble_evaluator import EvaluatorServerConfig
 from ert.run_models.everest_run_model import EverestRunModel
-from everest.config import EverestConfig
-from everest.plugins.site_config_env import PluginSiteConfigEnv
-from everest.simulator.everest_to_ert import everest_to_ert_config
-from everest.strings import EVEREST
-from everest.util import makedirs_if_needed
+
+from .config import EverestConfig
+from .plugins.site_config_env import PluginSiteConfigEnv
+from .simulator.everest_to_ert import everest_to_ert_config
+from .strings import EVEREST
+from .util import makedirs_if_needed
 
 
 def start_optimization(