One more bug fix for iterate option in BACKGROUNDS

deepskies · Dec 17, 2021 · 530539b · 530539b
1 parent efd1e6a
commit 530539b
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/HISTORY.rst b/HISTORY.rst
@@ -3,6 +3,10 @@
 History
 -------
 
+0.0.2.3 (2021-12-20)
++++++++++++++++++++++
+* Bug fixes for ITERATE option in image BACKGROUNDS
+
 0.0.2.2 (2021-12-13)
 +++++++++++++++++++++
 * Bug fixes for ITERATE option in image BACKGROUNDS

diff --git a/README.md b/README.md
@@ -2,7 +2,7 @@
 
 [![status](https://joss.theoj.org/papers/e978dd566d1f290055a02d76288e95e1/status.svg)](https://joss.theoj.org/papers/e978dd566d1f290055a02d76288e95e1)
 [![status](https://img.shields.io/badge/arXiv-2102.02830-red)](http://arxiv.org/abs/2102.02830)
-[![status](https://img.shields.io/badge/PyPi-0.0.2.2-blue)](https://pypi.org/project/deeplenstronomy/)
+[![status](https://img.shields.io/badge/PyPi-0.0.2.3-blue)](https://pypi.org/project/deeplenstronomy/)
 [![status](https://img.shields.io/badge/License-MIT-lightgrey)](https://github.com/deepskies/deeplenstronomy/blob/master/LICENSE)
 
 `deeplenstronomy` is a tool for simulating large datasets for applying deep learning to strong gravitational lensing. 

diff --git a/deeplenstronomy/deeplenstronomy.py b/deeplenstronomy/deeplenstronomy.py
@@ -419,7 +419,7 @@ def make_dataset(config, dataset=None, save_to_disk=False, store_in_memory=True,
         # Handle image backgrounds if they exist
         real_image_indices = []
         if len(parser.image_paths) > 0 and configuration in parser.image_configurations:
-            image_indices = organize_image_backgrounds(im_dir, len(image_backgrounds), [_flatten_image_info(sim_input) for sim_input in sim_inputs], configuration)
+            image_indices = organize_image_backgrounds(im_dir, len(image_backgrounds), [_flatten_image_info(sim_input) for sim_input in sim_inputs], configuration, overwrite=background_iterate)
             check_background_indices(image_indices, background_iterate)
         else:
             image_indices = np.zeros(len(sim_inputs), dtype=int)

diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 
 setup(
     name='deeplenstronomy',
-    version='0.0.2.2',
+    version='0.0.2.3',
     description='wrap lenstronomy for efficient simulation generation',
     long_description=long_description,
     long_description_content_type='text/markdown',