Skip to content

Commit 5928ce5

Browse files
Jammy2211claude
authored andcommitted
Fix galaxies_via_instance_from to include scaling_galaxies
Replace early-return + operator merge with list accumulation so that both extra_galaxies and scaling_galaxies are always added to the galaxy list. The old pattern caused scaling_galaxies to be silently dropped when extra_galaxies was also present (integer key collision in af.Collection +). Same fix applied to aggregator/galaxies.py _galaxies_from. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 13f65ce commit 5928ce5

2 files changed

Lines changed: 12 additions & 4 deletions

File tree

autogalaxy/aggregator/galaxies.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,10 @@ def _galaxies_from(fit: af.Fit, instance: af.ModelInstance) -> List[Galaxy]:
6868
if hasattr(instance, "extra_galaxies"):
6969
if instance.extra_galaxies is not None:
7070
galaxies = galaxies + instance.extra_galaxies
71+
72+
if hasattr(instance, "scaling_galaxies"):
73+
if instance.scaling_galaxies is not None:
74+
galaxies = galaxies + instance.scaling_galaxies
7175

7276
galaxies_list.append(galaxies)
7377

autogalaxy/analysis/analysis/analysis.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -77,13 +77,17 @@ def galaxies_via_instance_from(
7777
-------
7878
A list of galaxies that is used to then fit the dataset.
7979
"""
80+
galaxy_list = list(instance.galaxies)
81+
8082
if hasattr(instance, "extra_galaxies"):
8183
if getattr(instance, "extra_galaxies", None) is not None:
82-
return Galaxies(
83-
galaxies=instance.galaxies + instance.extra_galaxies,
84-
)
84+
galaxy_list += list(instance.extra_galaxies)
85+
86+
if hasattr(instance, "scaling_galaxies"):
87+
if getattr(instance, "scaling_galaxies", None) is not None:
88+
galaxy_list += list(instance.scaling_galaxies)
8589

86-
return Galaxies(galaxies=instance.galaxies)
90+
return Galaxies(galaxies=galaxy_list)
8791

8892
def dataset_model_via_instance_from(
8993
self, instance: af.ModelInstance

0 commit comments

Comments
 (0)