Skip to content

Commit dbfefe4

Browse files
Jammy2211claude
authored andcommitted
Fix _tracer_from aggregator to include scaling_galaxies
Replace Collection + merge with list accumulation so extra_galaxies and scaling_galaxies are both included without integer-key collision. Subhalo mutations now go to instance.galaxies before list() conversion, matching the pattern in analysis/analysis/lens.py. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 382b20a commit dbfefe4

1 file changed

Lines changed: 13 additions & 9 deletions

File tree

autolens/aggregator/tracer.py

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -65,12 +65,6 @@ def _tracer_from(
6565

6666
for instance in instance_list:
6767

68-
galaxies = instance.galaxies
69-
70-
if hasattr(instance, "extra_galaxies"):
71-
if instance.extra_galaxies is not None:
72-
galaxies = galaxies + instance.extra_galaxies
73-
7468
try:
7569
cosmology = instance.cosmology
7670
except AttributeError:
@@ -82,7 +76,7 @@ def _tracer_from(
8276
# TODO : These are ugly as hell (>_<)
8377

8478
if hasattr(instance, "perturb"):
85-
galaxies.subhalo = instance.perturb
79+
instance.galaxies.subhalo = instance.perturb
8680

8781
if hasattr(instance.galaxies, "subhalo"):
8882
subhalo_centre = tracer_util.grid_2d_at_redshift_from(
@@ -92,9 +86,19 @@ def _tracer_from(
9286
cosmology=cosmology,
9387
)
9488

95-
galaxies.subhalo.mass.centre = tuple(subhalo_centre.in_list[0])
89+
instance.galaxies.subhalo.mass.centre = tuple(subhalo_centre.in_list[0])
90+
91+
galaxy_list = list(instance.galaxies)
92+
93+
if hasattr(instance, "extra_galaxies"):
94+
if instance.extra_galaxies is not None:
95+
galaxy_list += list(instance.extra_galaxies)
96+
97+
if hasattr(instance, "scaling_galaxies"):
98+
if instance.scaling_galaxies is not None:
99+
galaxy_list += list(instance.scaling_galaxies)
96100

97-
tracer = Tracer(galaxies=galaxies, cosmology=cosmology)
101+
tracer = Tracer(galaxies=galaxy_list, cosmology=cosmology)
98102

99103
tracer_list.append(tracer)
100104

0 commit comments

Comments
 (0)