Change order of merging

src/graphnet/data/extractors/prometheus/pulsemap_simulator.py

@@ -175,19 +175,19 @@ def __call__(self, event: pd.DataFrame) -> pd.DataFrame:
                         photons=photons, column=self._time_column
                     )
 
+                    # Merge photons into pulses, add charge & delete photons
+                    pulses = self._merge_into_pulses(photons=photons)
+                    del photons  # save memory
+
                     # Smear arrival time
                     if self._time_std > 0:
-                        photons[self._time_column] = abs(
+                        pulses[self._time_column] = abs(
                             self._smear_feature(
-                                x=photons[self._time_column],
+                                x=pulses[self._time_column],
                                 std=self._time_std,
                             )
                         ).tolist()
 
-                    # Merge photons into pulses, add charge & delete photons
-                    pulses = self._merge_into_pulses(photons=photons)
-                    del photons  # save memory
-
                     # Smear Charge
                     if self._charge_std > 0:
                         pulses["charge"] = abs(