use absolute imports

Author: lkstrp

README.md

@@ -90,4 +90,4 @@ To generate a PDF of the dependency graph of all steps `build/dag.pdf` run:
 
 ## License
 
-The code in this repo is MIT licensed, see `./LICENSE.md`.
+The code in this repo is MIT licensed, see `./LICENSE.md`.

rules/common.smk

@@ -7,10 +7,9 @@ from functools import partial, lru_cache
 
 import os, sys, glob
 
-path = workflow.source_path("../scripts/_helpers.py")
-sys.path.insert(0, os.path.dirname(path))
+sys.path.insert(0, os.path.abspath("../scripts"))
 
-from _helpers import validate_checksum, update_config_from_wildcards
+from scripts._helpers import validate_checksum, update_config_from_wildcards
 from snakemake.utils import update_config
 
 

scripts/_benchmark.py

@@ -2,9 +2,7 @@
 # SPDX-FileCopyrightText: : 2020-2024 The PyPSA-Eur Authors
 #
 # SPDX-License-Identifier: MIT
-"""
-
-"""
+""" """
 
 from __future__ import absolute_import, print_function
 

scripts/add_brownfield.py

@@ -12,14 +12,15 @@
 import pandas as pd
 import pypsa
 import xarray as xr
-from _helpers import (
+from add_existing_baseyear import add_build_year_to_new_assets
+from pypsa.clustering.spatial import normed_or_uniform
+
+from scripts._helpers import (
     configure_logging,
     get_snapshots,
     set_scenario_config,
     update_config_from_wildcards,
 )
-from add_existing_baseyear import add_build_year_to_new_assets
-from pypsa.clustering.spatial import normed_or_uniform
 
 logger = logging.getLogger(__name__)
 idx = pd.IndexSlice
@@ -258,7 +259,7 @@ def update_heat_pump_efficiency(n: pypsa.Network, n_p: pypsa.Network, year: int)
 
 if __name__ == "__main__":
     if "snakemake" not in globals():
-        from _helpers import mock_snakemake
+        from scripts._helpers import mock_snakemake
 
         snakemake = mock_snakemake(
             "add_brownfield",

scripts/add_electricity.py

@@ -119,14 +119,15 @@
 import powerplantmatching as pm
 import pypsa
 import xarray as xr
-from _helpers import (
+from powerplantmatching.export import map_country_bus
+from pypsa.clustering.spatial import DEFAULT_ONE_PORT_STRATEGIES, normed_or_uniform
+
+from scripts._helpers import (
     configure_logging,
     get_snapshots,
     set_scenario_config,
     update_p_nom_max,
 )
-from powerplantmatching.export import map_country_bus
-from pypsa.clustering.spatial import DEFAULT_ONE_PORT_STRATEGIES, normed_or_uniform
 
 idx = pd.IndexSlice
 
@@ -308,7 +309,6 @@ def load_and_aggregate_powerplants(
     aggregation_strategies: dict = None,
     exclude_carriers: list = None,
 ) -> pd.DataFrame:
-
     if not aggregation_strategies:
         aggregation_strategies = {}
 
@@ -410,7 +410,6 @@ def attach_load(
     busmap_fn: str,
     scaling: float = 1.0,
 ) -> None:
-
     load = (
         xr.open_dataarray(load_fn).to_dataframe().squeeze(axis=1).unstack(level="time")
     )
@@ -431,7 +430,6 @@ def set_transmission_costs(
     line_length_factor: float = 1.0,
     link_length_factor: float = 1.0,
 ) -> None:
-
     n.lines["capital_cost"] = (
         n.lines["length"]
         * line_length_factor
@@ -1014,7 +1012,7 @@ def attach_stores(n, costs, extendable_carriers):
 
 if __name__ == "__main__":
     if "snakemake" not in globals():
-        from _helpers import mock_snakemake
+        from scripts._helpers import mock_snakemake
 
         snakemake = mock_snakemake("add_electricity", clusters=100)
     configure_logging(snakemake)

scripts/add_existing_baseyear.py

@@ -18,16 +18,21 @@
 import powerplantmatching as pm
 import pypsa
 import xarray as xr
-from _helpers import (
+
+from scripts._helpers import (
     configure_logging,
     set_scenario_config,
     update_config_from_wildcards,
 )
-from add_electricity import sanitize_carriers
-from definitions.heat_sector import HeatSector
-from definitions.heat_system import HeatSystem
-from definitions.heat_system_type import HeatSystemType
-from prepare_sector_network import cluster_heat_buses, define_spatial, prepare_costs
+from scripts.add_electricity import sanitize_carriers
+from scripts.definitions.heat_sector import HeatSector
+from scripts.definitions.heat_system import HeatSystem
+from scripts.definitions.heat_system_type import HeatSystemType
+from scripts.prepare_sector_network import (
+    cluster_heat_buses,
+    define_spatial,
+    prepare_costs,
+)
 
 logger = logging.getLogger(__name__)
 cc = coco.CountryConverter()
@@ -1063,7 +1068,7 @@ def set_defaults(n):
 # %%
 if __name__ == "__main__":
     if "snakemake" not in globals():
-        from _helpers import mock_snakemake
+        from scripts._helpers import mock_snakemake
 
         snakemake = mock_snakemake(
             "add_existing_baseyear",

scripts/add_transmission_projects_and_dlr.py

@@ -13,7 +13,8 @@
 import pandas as pd
 import pypsa
 import xarray as xr
-from _helpers import configure_logging, set_scenario_config
+
+from scripts._helpers import configure_logging, set_scenario_config
 
 logger = logging.getLogger(__name__)
 
@@ -71,7 +72,7 @@ def attach_line_rating(
 
 if __name__ == "__main__":
     if "snakemake" not in globals():
-        from _helpers import mock_snakemake
+        from scripts._helpers import mock_snakemake
 
         snakemake = mock_snakemake("add_transmission_projects_and_dlr")
     configure_logging(snakemake)
@@ -82,11 +83,9 @@ def attach_line_rating(
     n = pypsa.Network(snakemake.input.network)
 
     if params["transmission_projects"]["enable"]:
-
         attach_transmission_projects(n, snakemake.input.transmission_projects)
 
     if params["dlr"]["activate"]:
-
         rating = xr.open_dataarray(snakemake.input.dlr).to_pandas().transpose()
 
         s_max_pu = params["s_max_pu"]

scripts/base_network.py

@@ -86,14 +86,19 @@
 import shapely.prepared
 import shapely.wkt
 import yaml
-from _helpers import REGION_COLS, configure_logging, get_snapshots, set_scenario_config
 from packaging.version import Version, parse
 from scipy.sparse import csgraph
 from scipy.spatial import KDTree
 from shapely.geometry import LineString, Point
 
-PD_GE_2_2 = parse(pd.__version__) >= Version("2.2")
+from scripts._helpers import (
+    REGION_COLS,
+    configure_logging,
+    get_snapshots,
+    set_scenario_config,
+)
 
+PD_GE_2_2 = parse(pd.__version__) >= Version("2.2")
 logger = logging.getLogger(__name__)
 
 
@@ -976,7 +981,7 @@ def append_bus_shapes(n, shapes, type):
 
 if __name__ == "__main__":
     if "snakemake" not in globals():
-        from _helpers import mock_snakemake
+        from scripts._helpers import mock_snakemake
 
         snakemake = mock_snakemake("base_network")
     configure_logging(snakemake)

scripts/build_ammonia_production.py

@@ -25,14 +25,15 @@
 import country_converter as coco
 import numpy as np
 import pandas as pd
-from _helpers import set_scenario_config
+
+from scripts._helpers import set_scenario_config
 
 cc = coco.CountryConverter()
 
 
 if __name__ == "__main__":
     if "snakemake" not in globals():
-        from _helpers import mock_snakemake
+        from scripts._helpers import mock_snakemake
 
         snakemake = mock_snakemake("build_ammonia_production")
 

scripts/build_biomass_potentials.py

@@ -12,8 +12,9 @@
 import geopandas as gpd
 import numpy as np
 import pandas as pd
-from _helpers import configure_logging, set_scenario_config
-from build_energy_totals import build_eurostat
+
+from scripts._helpers import configure_logging, set_scenario_config
+from scripts.build_energy_totals import build_eurostat
 
 logger = logging.getLogger(__name__)
 AVAILABLE_BIOMASS_YEARS = [2010, 2020, 2030, 2040, 2050]
@@ -339,8 +340,7 @@ def add_unsustainable_potentials(df):
 
 if __name__ == "__main__":
     if "snakemake" not in globals():
-
-        from _helpers import mock_snakemake
+        from scripts._helpers import mock_snakemake
 
         snakemake = mock_snakemake(
             "build_biomass_potentials",