Cosmetic changes.

NKI-AI · Aug 16, 2020 · 58d39cc · 58d39cc
1 parent 208615e
commit 58d39cc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/direct/data/datasets.py b/direct/data/datasets.py
@@ -69,7 +69,9 @@ def __getitem__(self, idx: int) -> Dict[str, Any]:
             # Some images have strange behavior.
             image_shape = sample["kspace"].shape
 
-            if image_shape[-1] < sample["reconstruction_size"][-2]:  # reconstruction size is (x, y, z)
+            if (
+                image_shape[-1] < sample["reconstruction_size"][-2]
+            ):  # reconstruction size is (x, y, z)
                 warnings.warn(
                     f"Encountered {sample['filename']} with header reconstruction size {sample['reconstruction_size']}, "
                     f" yet matrix size is {image_shape}, this is a known issue in the FastMRI dataset."

diff --git a/direct/nn/rim/rim_engine.py b/direct/nn/rim/rim_engine.py
@@ -197,8 +197,10 @@ def evaluate(
             data = AddNames()(data)
             filenames = data.pop("filename")
             if len(set(filenames)) != 1:
-                raise ValueError(f"Expected a batch during validation to only contain filenames of one case. "
-                                 f"Got {set(filenames)}.")
+                raise ValueError(
+                    f"Expected a batch during validation to only contain filenames of one case. "
+                    f"Got {set(filenames)}."
+                )
 
             slice_nos = data.pop("slice_no")
             scaling_factors = data.pop("scaling_factor")