improve typing

Signed-off-by: Victor Garcia Reolid <victor@seita.nl>
FlexMeasures · Dec 16, 2023 · b3ac617 · b3ac617
1 parent 8412291
commit b3ac617
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/flexmeasures/ui/crud/assets.py b/flexmeasures/ui/crud/assets.py
@@ -152,7 +152,7 @@ def expunge_asset():
                 Sensor.generic_asset_id == asset_data["id"]
             ).all()
             expunge_asset()
-        if asset_data.get("parent_asset_id") is not None:
+        if asset_data.get("parent_asset_id", None) is not None:
             asset.parent_asset = GenericAsset.query.filter(
                 GenericAsset.id == asset_data["parent_asset_id"]
             ).one_or_none()

diff --git a/flexmeasures/ui/utils/breadcrumb_utils.py b/flexmeasures/ui/utils/breadcrumb_utils.py
@@ -10,7 +10,7 @@ def get_breadcrumb_info(entity: Sensor | Asset | Account | None) -> dict:
     }
 
 
-def get_ancestry(entity: Sensor | Asset | Account | None) -> list:
+def get_ancestry(entity: Sensor | Asset | Account | None) -> list[dict]:
 
     # Public account
     if entity is None:

diff --git a/flexmeasures/ui/utils/view_utils.py b/flexmeasures/ui/utils/view_utils.py
@@ -81,7 +81,7 @@ def render_flexmeasures_template(html_filename: str, **variables):
     variables["extra_css"] = current_app.config.get("FLEXMEASURES_EXTRA_CSS_PATH")
 
     if "asset" in variables:
-        variables["breadcrumb_info"] = (get_breadcrumb_info(asset),)
+        variables["breadcrumb_info"] = get_breadcrumb_info(asset)
 
     return render_template(html_filename, **variables)