From 8dfab454537fa3892cce1a939625e671aca3091e Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 27 Feb 2024 12:35:55 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 rules/retrieve.smk                    |  7 +++++--
 scripts/retrieve_eurostat_data.py     | 14 ++++++--------
 scripts/retrieve_sector_databundle.py |  2 +-
 3 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/rules/retrieve.smk b/rules/retrieve.smk
index 9250e745..cfe698fc 100644
--- a/rules/retrieve.smk
+++ b/rules/retrieve.smk
@@ -160,14 +160,17 @@ if config["enable"]["retrieve"] and config["enable"].get(
     rule retrieve_eurostat_data:
         output:
             protected(
-            directory("data/bundle-sector/eurostat-energy_balances-april_2023_edition")
-        ),
+                directory(
+                    "data/bundle-sector/eurostat-energy_balances-april_2023_edition"
+                )
+            ),
         log:
             "logs/retrieve_eurostat_data.log",
         retries: 2
         script:
             "../scripts/retrieve_eurostat_data.py"
 
+
 if config["enable"]["retrieve"]:
     datafiles = [
         "IGGIELGN_LNGs.geojson",
diff --git a/scripts/retrieve_eurostat_data.py b/scripts/retrieve_eurostat_data.py
index aedc43c9..daee4fc4 100644
--- a/scripts/retrieve_eurostat_data.py
+++ b/scripts/retrieve_eurostat_data.py
@@ -11,11 +11,7 @@ import logging
 import zipfile
 from pathlib import Path
 
-from _helpers import (
-    configure_logging,
-    progress_retrieve,
-    set_scenario_config,
-)
+from _helpers import configure_logging, progress_retrieve, set_scenario_config
 
 logger = logging.getLogger(__name__)
 
@@ -33,13 +29,15 @@ if __name__ == "__main__":
     disable_progress = snakemake.config["run"].get("disable_progressbar", False)
     url_eurostat = "https://ec.europa.eu/eurostat/documents/38154/4956218/Balances-December2022.zip/f7cf0d19-5c0f-60ad-4e48-098a5ddd6e48?t=1671184070589"
     tarball_fn = Path(f"{rootpath}/data/bundle-sector/eurostat_2023.zip")
-    to_fn = Path(f"{rootpath}/data/bundle-sector/eurostat-energy_balances-april_2023_edition/")
+    to_fn = Path(
+        f"{rootpath}/data/bundle-sector/eurostat-energy_balances-april_2023_edition/"
+    )
 
     logger.info(f"Downloading Eurostat data from '{url_eurostat}'.")
     progress_retrieve(url_eurostat, tarball_fn, disable=disable_progress)
 
     logger.info("Extracting Eurostat data.")
-    with zipfile.ZipFile(tarball_fn, 'r') as zip_ref:
+    with zipfile.ZipFile(tarball_fn, "r") as zip_ref:
         zip_ref.extractall(to_fn)
 
-    logger.info(f"Eurostat data available in '{to_fn}'.")
\ No newline at end of file
+    logger.info(f"Eurostat data available in '{to_fn}'.")
diff --git a/scripts/retrieve_sector_databundle.py b/scripts/retrieve_sector_databundle.py
index d311dac7..3b825da2 100644
--- a/scripts/retrieve_sector_databundle.py
+++ b/scripts/retrieve_sector_databundle.py
@@ -46,4 +46,4 @@ if __name__ == "__main__":
 
     tarball_fn.unlink()
 
-    logger.info(f"Databundle available in '{to_fn}'.")
\ No newline at end of file
+    logger.info(f"Databundle available in '{to_fn}'.")