bjoern/pypsa-eur
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

make set_scenario_config robust against mock_snakemake and subworkflow

Browse Source
This commit is contained in:
Fabian 2023-08-21 12:24:27 +02:00
parent ebb5da655e
commit 1810bbd4b3
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
scripts/_helpers.py
View File

@ -31,11 +31,16 @@ def mute_print():
def set_scenario_config(snakemake):
if snakemake.config["run"]["scenarios"]:
script_dir = Path(__file__).parent.resolve()
root_dir = script_dir.parent
with open(root_dir / snakemake.config["scenariofile"], "r") as f:
scenario_config = yaml.safe_load(f)
if snakemake.config["run"]["scenarios"] and "run" in snakemake.wildcards:
try:
with open(snakemake.config["scenariofile"], "r") as f:
scenario_config = yaml.safe_load(f)
except FileNotFoundError:
# fallback for mock_snakemake
script_dir = Path(__file__).parent.resolve()
root_dir = script_dir.parent
with open(root_dir / snakemake.config["scenariofile"], "r") as f:
scenario_config = yaml.safe_load(f)
update_config(snakemake.config, scenario_config[snakemake.wildcards.run])