add type hints

Signed-off-by: Thijs Baaijen <[email protected]>

Author: Thijss

src/power_grid_model_ds/_core/visualizer/callbacks/element_scaling.py

@@ -3,6 +3,7 @@
 # SPDX-License-Identifier: MPL-2.0
 
 from copy import deepcopy
+from typing import Any
 
 from dash import Input, Output, callback
 
@@ -15,7 +16,7 @@
     Input("edge-scale-input", "value"),
     prevent_initial_call=True,
 )
-def scale_elements(node_scale, edge_scale):
+def scale_elements(node_scale: float, edge_scale: float) -> list[dict[str, Any]]:
     """Callback to scale the elements of the graph."""
     new_stylesheet = deepcopy(DEFAULT_STYLESHEET)
     edge_style = {

src/power_grid_model_ds/_core/visualizer/callbacks/search_form.py

@@ -1,6 +1,7 @@
 # SPDX-FileCopyrightText: Contributors to the Power Grid Model project <[email protected]>
 #
 # SPDX-License-Identifier: MPL-2.0
+from typing import Any
 
 from dash import Input, Output, callback
 
@@ -15,7 +16,7 @@
     Input("search-form-operator-input", "value"),
     Input("search-form-value-input", "value"),
 )
-def search_element(group, column, operator, value):
+def search_element(group: str, column: str, operator: str, value: str) -> list[dict[str, Any]]:
     """Color the specified element red based on the input values."""
     if not group or not column or not value:
         return DEFAULT_STYLESHEET