linting fixes

leifdenby · Aug 23, 2022 · 75eb7a1 · 75eb7a1
1 parent d97158d
commit 75eb7a1
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git 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
@@ -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 = {}