diff --git a/scripts/_helpers.py b/scripts/_helpers.py index 1c890ea5..d97144ba 100644 --- a/scripts/_helpers.py +++ b/scripts/_helpers.py @@ -786,6 +786,3 @@ def get_snapshots(snapshots, drop_leap_day=False, freq="h", **kwargs): return time - -def get_country_from_node_name(node_name: str) -> str: - return node_name[:2] diff --git a/scripts/build_cop_profiles/run.py b/scripts/build_cop_profiles/run.py index 23229e98..401fdc18 100644 --- a/scripts/build_cop_profiles/run.py +++ b/scripts/build_cop_profiles/run.py @@ -46,7 +46,7 @@ import geopandas as gpd import numpy as np import pandas as pd import xarray as xr -from _helpers import get_country_from_node_name, set_scenario_config +from _helpers import set_scenario_config from CentralHeatingCopApproximator import CentralHeatingCopApproximator from DecentralHeatingCopApproximator import DecentralHeatingCopApproximator @@ -134,6 +134,8 @@ def get_cop( source_type=heat_source, ).approximate_cop() +def get_country_from_node_name(node_name: str) -> str: + return node_name[:2] if __name__ == "__main__": if "snakemake" not in globals():