From 75eb7a14e198464eaa309ecd5c904ee76a88be21 Mon Sep 17 00:00:00 2001
From: Leif Denby <leif@denby.eu>
Date: Tue, 23 Aug 2022 09:16:23 +0100
Subject: [PATCH] linting fixes

---
 genesis/utils/pipeline/data/extraction.py       | 17 +++++++++++++++--
 .../utils/pipeline/data_sources/uclales/base.py |  6 ++++--
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/genesis/utils/pipeline/data/extraction.py b/genesis/utils/pipeline/data/extraction.py
index db71a03..d6a2e66 100644
--- a/genesis/utils/pipeline/data/extraction.py
+++ b/genesis/utils/pipeline/data/extraction.py
@@ -221,7 +221,13 @@ def requires(self):
         # TODO remove hardcoded orientation
         base_name = meta.get("experiment_name", self.base_name)
         dest_path = get_workdir() / self.base_name / "cross_sections" / "runtime_slices"
-        task = fn(dataset_meta=meta, var_name=self.var_name, orientation="xy", dest_path=str(dest_path), base_name=base_name)
+        task = fn(
+            dataset_meta=meta,
+            var_name=self.var_name,
+            orientation="xy",
+            dest_path=str(dest_path),
+            base_name=base_name,
+        )
         return task
 
     def _extract_and_symlink_local_file(self):
@@ -395,7 +401,14 @@ def output(self):
 
         fn = ".".join(name_parts)
 
-        p = get_workdir() / self.base_name / "cross_sections" / "runtime_slices" / "by_time" / fn
+        p = (
+            get_workdir()
+            / self.base_name
+            / "cross_sections"
+            / "runtime_slices"
+            / "by_time"
+            / fn
+        )
         return XArrayTarget(str(p))
 
 
diff --git a/genesis/utils/pipeline/data_sources/uclales/base.py b/genesis/utils/pipeline/data_sources/uclales/base.py
index 60b7e98..266aa41 100644
--- a/genesis/utils/pipeline/data_sources/uclales/base.py
+++ b/genesis/utils/pipeline/data_sources/uclales/base.py
@@ -128,7 +128,7 @@ def _build_block_extraction_task(dataset_meta, field_name):
         source_path=raw_data_path,
         file_prefix=dataset_meta["experiment_name"],
         tn=dataset_meta["timestep"],
-        kind="3d"
+        kind="3d",
     )
 
     if field_name in ["u", "v", "w"]:
@@ -143,7 +143,9 @@ def _build_block_extraction_task(dataset_meta, field_name):
     return task
 
 
-def build_runtime_cross_section_extraction_task(dataset_meta, var_name, orientation, base_name, dest_path):
+def build_runtime_cross_section_extraction_task(
+    dataset_meta, var_name, orientation, base_name, dest_path
+):
     raw_data_path = Path(dataset_meta["path"]) / "raw_data"
 
     task_kwargs = {}