Merge pull request #133 from automl/fix-config-usage

sarah-segel · web-flow · commit 71ad083dad1b · 2024-05-07T08:59:31.000+02:00
Fix use of config
diff --git a/deepcave/__init__.py b/deepcave/__init__.py
@@ -137,15 +137,24 @@ def get_app(title: str) -> Any:
     try:
         from deepcave.runs.objective import Objective  # noqa
         from deepcave.runs.recorder import Recorder  # noqa
+        from deepcave.utils.configs import parse_config
+        from deepcave.utils.notification import Notification
 
-        __all__ = ["version", "Recorder", "Objective"]
+        config_name = None
+        if "--config" in sys.argv:
+            config_name = sys.argv[sys.argv.index("--config") + 1]
+        config = parse_config(config_name)
+
+        # Notifications
+        notification = Notification()
+
+        __all__ = ["version", "Recorder", "Objective", "notification", "config"]
     except ModuleNotFoundError:
         __all__ = ["version"]
 
 
 _api_mode = False if "app" in globals() else True
 
-
 # This TypeVar is necessary to ensure that the decorator works with arbitrary signatures.
 F = TypeVar("F", bound=Callable[..., Any])
 
diff --git a/deepcave/plugins/__init__.py b/deepcave/plugins/__init__.py
@@ -115,9 +115,9 @@ def get_base_url(cls) -> str:
         str
             Url for the plugin as string.
         """
-        from deepcave.config import Config
+        from deepcave import config
 
-        return f"http://{Config.DASH_ADDRESS}:{Config.DASH_PORT}/plugins/{cls.id}"
+        return f"http://{config.DASH_ADDRESS}:{config.DASH_PORT}/plugins/{cls.id}"
 
     @staticmethod
     def check_run_compatibility(run: AbstractRun) -> bool:
diff --git a/deepcave/plugins/budget/budget_correlation.py b/deepcave/plugins/budget/budget_correlation.py
@@ -19,8 +19,7 @@
 from dash import dcc, html
 from scipy import stats
 
-from deepcave import notification
-from deepcave.config import Config
+from deepcave import config, notification
 from deepcave.plugins.dynamic import DynamicPlugin
 from deepcave.runs import AbstractRun, Status
 from deepcave.utils.layout import create_table, get_select_options
@@ -229,9 +228,9 @@ def get_output_layout(register: Callable) -> List[Any]:
                     dbc.Tab(
                         dcc.Graph(
                             id=register("graph", "figure"),
-                            style={"height": Config.FIGURE_HEIGHT},
+                            style={"height": config.FIGURE_HEIGHT},
                             config={
-                                "toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}
+                                "toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}
                             },
                         ),
                         label="Graph",
@@ -310,7 +309,7 @@ def load_outputs(run, _, outputs) -> List[Any]:  # type: ignore
         layout = go.Layout(
             xaxis=dict(title="Budget"),
             yaxis=dict(title="Correlation"),
-            margin=Config.FIGURE_MARGIN,
+            margin=config.FIGURE_MARGIN,
             legend=dict(title="Budgets"),
         )
 
diff --git a/deepcave/plugins/hyperparameter/importances.py b/deepcave/plugins/hyperparameter/importances.py
@@ -20,7 +20,7 @@
 from dash import dcc, html
 from dash.exceptions import PreventUpdate
 
-from deepcave.config import Config
+from deepcave import config
 from deepcave.evaluators.fanova import fANOVA as GlobalEvaluator
 from deepcave.evaluators.lpi import LPI as LocalEvaluator
 from deepcave.plugins.static import StaticPlugin
@@ -359,8 +359,8 @@ def get_output_layout(register: Callable) -> dcc.Graph:
         """
         return dcc.Graph(
             register("graph", "figure"),
-            style={"height": Config.FIGURE_HEIGHT},
-            config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+            style={"height": config.FIGURE_HEIGHT},
+            config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
         )
 
     @staticmethod
@@ -456,7 +456,7 @@ def load_outputs(run, inputs, outputs) -> go.Figure:  # type: ignore
             barmode="group",
             yaxis_title="Importance",
             legend={"title": "Budget"},
-            margin=Config.FIGURE_MARGIN,
+            margin=config.FIGURE_MARGIN,
             xaxis=dict(tickangle=-45),
         )
         save_image(figure, "importances.pdf")
diff --git a/deepcave/plugins/hyperparameter/pdp.py b/deepcave/plugins/hyperparameter/pdp.py
@@ -25,7 +25,7 @@
 from dash import dcc, html
 from pyPDP.algorithms.pdp import PDP
 
-from deepcave.config import Config
+from deepcave import config
 from deepcave.evaluators.epm.random_forest_surrogate import RandomForestSurrogate
 from deepcave.plugins.static import StaticPlugin
 from deepcave.runs import Status
@@ -368,8 +368,8 @@ def get_output_layout(register: Callable) -> dcc.Graph:
         """
         return dcc.Graph(
             register("graph", "figure"),
-            style={"height": Config.FIGURE_HEIGHT},
-            config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+            style={"height": config.FIGURE_HEIGHT},
+            config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
         )
 
     @staticmethod
@@ -507,7 +507,7 @@ def get_pdp_figure(  # type: ignore
                 dict(
                     xaxis=dict(tickvals=x_tickvals, ticktext=x_ticktext, title=hp1_name),
                     yaxis=dict(tickvals=y_tickvals, ticktext=y_ticktext, title=hp2_name),
-                    margin=Config.FIGURE_MARGIN,
+                    margin=config.FIGURE_MARGIN,
                     title=title,
                 )
             )
diff --git a/deepcave/plugins/hyperparameter/symbolic_explanations.py b/deepcave/plugins/hyperparameter/symbolic_explanations.py
@@ -27,7 +27,7 @@
 from gplearn.genetic import SymbolicRegressor
 from pyPDP.algorithms.pdp import PDP
 
-from deepcave.config import Config
+from deepcave import config
 from deepcave.evaluators.epm.random_forest_surrogate import RandomForestSurrogate
 from deepcave.plugins.hyperparameter.pdp import PartialDependencies
 from deepcave.plugins.static import StaticPlugin
@@ -493,13 +493,13 @@ def get_output_layout(register: Callable) -> List[dcc.Graph]:
         return [
             dcc.Graph(
                 register("symb_graph", "figure"),
-                style={"height": Config.FIGURE_HEIGHT},
-                config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+                style={"height": config.FIGURE_HEIGHT},
+                config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
             ),
             dcc.Graph(
                 register("pdp_graph", "figure"),
-                style={"height": Config.FIGURE_HEIGHT},
-                config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+                style={"height": config.FIGURE_HEIGHT},
+                config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
             ),
         ]
 
@@ -600,7 +600,7 @@ def load_outputs(run, inputs, outputs) -> List[go.Figure]:  # type: ignore
                 dict(
                     xaxis=dict(tickvals=x_tickvals, ticktext=x_ticktext, title=hp1_name),
                     yaxis=dict(tickvals=y_tickvals, ticktext=y_ticktext, title=hp2_name),
-                    margin=Config.FIGURE_MARGIN,
+                    margin=config.FIGURE_MARGIN,
                     title=expr,
                 )
             )
diff --git a/deepcave/plugins/objective/configuration_cube.py b/deepcave/plugins/objective/configuration_cube.py
@@ -19,7 +19,7 @@
 from dash import dcc, html
 from dash.exceptions import PreventUpdate
 
-from deepcave.config import Config
+from deepcave import config
 from deepcave.plugins.dynamic import DynamicPlugin
 from deepcave.runs import AbstractRun, Status
 from deepcave.utils.compression import deserialize, serialize
@@ -297,8 +297,8 @@ def get_output_layout(register: Callable) -> Tuple[dcc.Graph,]:
         return (
             dcc.Graph(
                 register("graph", "figure"),
-                style={"height": Config.FIGURE_HEIGHT},
-                config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+                style={"height": config.FIGURE_HEIGHT},
+                config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
             ),
         )
 
@@ -428,7 +428,7 @@ def load_outputs(run, inputs, outputs) -> go.Figure:  # type: ignore
             layout = go.Layout(**layout_kwargs)
 
         figure = go.Figure(data=trace, layout=layout)
-        figure.update_layout(dict(margin=Config.FIGURE_MARGIN))
+        figure.update_layout(dict(margin=config.FIGURE_MARGIN))
         save_image(figure, "configuration_cube.pdf")
 
         return figure
diff --git a/deepcave/plugins/objective/cost_over_time.py b/deepcave/plugins/objective/cost_over_time.py
@@ -19,8 +19,7 @@
 from dash import dcc, html
 from dash.exceptions import PreventUpdate
 
-from deepcave import notification
-from deepcave.config import Config
+from deepcave import config, notification
 from deepcave.plugins.dynamic import DynamicPlugin
 from deepcave.runs import AbstractRun, check_equality
 from deepcave.runs.exceptions import NotMergeableError, RunInequality
@@ -300,8 +299,8 @@ def get_output_layout(register: Callable) -> dcc.Graph:
         """
         return dcc.Graph(
             register("graph", "figure"),
-            style={"height": Config.FIGURE_HEIGHT},
-            config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+            style={"height": config.FIGURE_HEIGHT},
+            config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
         )
 
     @staticmethod
@@ -423,7 +422,7 @@ def load_outputs(runs, inputs, outputs) -> go.Figure:  # type: ignore
         layout = go.Layout(
             xaxis=dict(title=xaxis_label, type=type),
             yaxis=dict(title=objective.name),
-            margin=Config.FIGURE_MARGIN,
+            margin=config.FIGURE_MARGIN,
         )
 
         figure = go.Figure(data=traces, layout=layout)
diff --git a/deepcave/plugins/objective/parallel_coordinates.py b/deepcave/plugins/objective/parallel_coordinates.py
@@ -20,7 +20,7 @@
 from dash import dcc, html
 from dash.exceptions import PreventUpdate
 
-from deepcave.config import Config
+from deepcave import config
 from deepcave.constants import VALUE_RANGE
 from deepcave.evaluators.fanova import fANOVA
 from deepcave.plugins.static import StaticPlugin
@@ -326,8 +326,8 @@ def get_output_layout(register: Callable) -> dcc.Graph:
         """
         return dcc.Graph(
             register("graph", "figure"),
-            style={"height": Config.FIGURE_HEIGHT},
-            config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+            style={"height": config.FIGURE_HEIGHT},
+            config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
         )
 
     @staticmethod
diff --git a/deepcave/plugins/objective/pareto_front.py b/deepcave/plugins/objective/pareto_front.py
@@ -17,8 +17,7 @@
 import plotly.graph_objs as go
 from dash import dcc, html
 
-from deepcave import notification
-from deepcave.config import Config
+from deepcave import config, notification
 from deepcave.plugins.dynamic import DynamicPlugin
 from deepcave.runs import AbstractRun, Status, check_equality
 from deepcave.runs.exceptions import NotMergeableError, RunInequality
@@ -381,8 +380,8 @@ def get_output_layout(register: Callable) -> dcc.Graph:
         """
         return dcc.Graph(
             register("graph", "figure"),
-            style={"height": Config.FIGURE_HEIGHT},
-            config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+            style={"height": config.FIGURE_HEIGHT},
+            config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
         )
 
     @staticmethod
@@ -524,7 +523,7 @@ def load_outputs(runs, inputs, outputs) -> go.Figure:  # type: ignore
             layout = go.Layout(
                 xaxis=dict(title=objective_1.name),
                 yaxis=dict(title=objective_2.name),
-                margin=Config.FIGURE_MARGIN,
+                margin=config.FIGURE_MARGIN,
             )
         else:
             layout = None
diff --git a/deepcave/plugins/static.py b/deepcave/plugins/static.py
@@ -389,12 +389,12 @@ def __call__(self) -> List[Component]:  # type: ignore
         List[Component]
             Layout as list of components.
         """
-        from deepcave.config import Config
+        from deepcave import config
 
         self._setup()
 
         components = [
-            dcc.Interval(id=self.get_internal_id("update-interval"), interval=Config.REFRESH_RATE),
+            dcc.Interval(id=self.get_internal_id("update-interval"), interval=config.REFRESH_RATE),
             dcc.Store(id=self.get_internal_id("update-interval-output"), data=0),
         ]
         components += super().__call__(True)
diff --git a/deepcave/plugins/summary/configurations.py b/deepcave/plugins/summary/configurations.py
@@ -20,7 +20,7 @@
 import plotly.graph_objs as go
 from dash import dcc, html
 
-from deepcave.config import Config
+from deepcave import config
 from deepcave.constants import VALUE_RANGE
 from deepcave.plugins.dynamic import DynamicPlugin
 from deepcave.runs import AbstractRun
@@ -304,9 +304,9 @@ def get_output_layout(
                     dbc.Tab(
                         dcc.Graph(
                             id=register("performance_graph", "figure"),
-                            style={"height": Config.FIGURE_HEIGHT},
+                            style={"height": config.FIGURE_HEIGHT},
                             config={
-                                "toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}
+                                "toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}
                             },
                         ),
                         label="Graph",
@@ -321,9 +321,9 @@ def get_output_layout(
                     dbc.Tab(
                         dcc.Graph(
                             id=register("configspace_graph", "figure"),
-                            style={"height": Config.FIGURE_HEIGHT},
+                            style={"height": config.FIGURE_HEIGHT},
                             config={
-                                "toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}
+                                "toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}
                             },
                         ),
                         label="Graph",
@@ -390,7 +390,7 @@ def _get_objective_figure(
             objective_data.append(trace)
 
         layout_kwargs: Dict[str, Any] = {
-            "margin": Config.FIGURE_MARGIN,
+            "margin": config.FIGURE_MARGIN,
             "xaxis": {"title": "Budget", "domain": [0.05 * len(run.get_objectives()), 1]},
         }
 
diff --git a/deepcave/plugins/summary/footprint.py b/deepcave/plugins/summary/footprint.py
@@ -16,7 +16,7 @@
 import plotly.graph_objs as go
 from dash import dcc, html
 
-from deepcave.config import Config
+from deepcave import config
 from deepcave.evaluators.footprint import Footprint as Evaluator
 from deepcave.plugins.static import StaticPlugin
 from deepcave.utils.layout import get_select_options, help_button
@@ -297,16 +297,16 @@ def get_output_layout(register: Callable) -> dbc.Tabs:
                 dbc.Tab(
                     dcc.Graph(
                         id=register("performance", "figure"),
-                        style={"height": Config.FIGURE_HEIGHT},
-                        config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+                        style={"height": config.FIGURE_HEIGHT},
+                        config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
                     ),
                     label="Performance",
                 ),
                 dbc.Tab(
                     dcc.Graph(
                         id=register("area", "figure"),
-                        style={"height": Config.FIGURE_HEIGHT},
-                        config={"toImageButtonOptions": {"scale": Config.FIGURE_DOWNLOAD_SCALE}},
+                        style={"height": config.FIGURE_HEIGHT},
+                        config={"toImageButtonOptions": {"scale": config.FIGURE_DOWNLOAD_SCALE}},
                     ),
                     label="Coverage",
                 ),
@@ -416,7 +416,7 @@ def load_outputs(run, inputs, outputs) -> List[Any]:  # type: ignore
         layout = go.Layout(
             xaxis=dict(title=None, tickvals=[]),
             yaxis=dict(title=None, tickvals=[]),
-            margin=Config.FIGURE_MARGIN,
+            margin=config.FIGURE_MARGIN,
         )
 
         performance = go.Figure(data=[performance_data] + traces, layout=layout)
diff --git a/deepcave/plugins/summary/overview.py b/deepcave/plugins/summary/overview.py
