Clean up

lanl · Jul 23, 2024 · cc678fd · cc678fd
1 parent aa1467a
commit cc678fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/scico/flax/examples/data_generation.py b/scico/flax/examples/data_generation.py
@@ -25,7 +25,7 @@
 else:
     have_ray_and_xdesign = True
     from .ray_functions import (
-        generate_foam1_images,  # noqa
+        generate_foam1_images,
         generate_foam2_images,
         distributed_data_generation,
     )
@@ -171,7 +171,7 @@ def generate_blur_data(
     size: int,
     blur_kernel: Array,
     noise_sigma: float,
-    imgfunc: Callable,
+    imgfunc: Callable = generate_foam1_images,
     seed: int = 4321,
     verbose: bool = False,
 ) -> Tuple[Array, Array]:

diff --git a/scico/flax/examples/examples.py b/scico/flax/examples/examples.py
@@ -16,7 +16,7 @@
 from scico.numpy import Array
 from scico.typing import Shape
 
-from .data_generation import generate_blur_data, generate_ct_data, generate_foam1_images
+from .data_generation import generate_blur_data, generate_ct_data
 from .data_preprocessing import ConfigImageSetDict, build_image_dataset, get_bsds_data
 from .typed_dict import CTDataSetDict
 
@@ -288,7 +288,6 @@ def load_blur_data(
         size,
         blur_kernel,
         noise_sigma,
-        imgfunc=generate_foam1_images,
         verbose=verbose,
     )
     # Separate training and testing partitions.