style: add return types of functions

Co-Authored-By: Francesco Sanvito <68587472+FraSanvit@users.noreply.github.com>
calliope-project · May 14, 2024 · 5d5b4eb · 5d5b4eb
1 parent 0832b0e
commit 5d5b4eb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/scripts/transport/road_transport_controlled_charging.py b/scripts/transport/road_transport_controlled_charging.py
@@ -45,7 +45,7 @@ def convert_annual_distance_to_electricity_demand(
     final_year: int,
     conversion_factors: dict[str, float],
     country_codes: list[str],
-):
+) -> pd.DataFrame:
     """
     Convert annual distance driven demand to electricity demand for
     controlled charging accounting for conversion factors.
@@ -73,7 +73,7 @@ def extract_national_ev_charging_potentials(
     conversion_factors: dict[str, float],
     battery_sizes: dict[str, float],
     country_codes: list[str],
-):
+) -> pd.DataFrame:
     # Extract number of EVs per vehicle type
     df_ev_numbers = (
         pd.read_csv(path_to_ev_numbers, index_col=[0, 1, 2, 3, 4])

diff --git a/scripts/transport/road_transport_controlled_constraints.py b/scripts/transport/road_transport_controlled_constraints.py
@@ -47,7 +47,7 @@ def get_national_ev_profiles(
     final_year: int,
     country_neighbour_dict: dict[str, list[str]],
     country_codes: list[str],
-):
+) -> pd.DataFrame:
     df_timeseries = (
         pd.read_csv(ev_profiles_path, index_col=[0, 1, 2], parse_dates=[0])
         .xs(slice(first_year, final_year), level="year")