Fix duplicated indices for augmented observations

kboone · May 13, 2019 · f462711 · f462711
1 parent 212eba6
commit f462711
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/avocado/astronomical_object.py b/avocado/astronomical_object.py
@@ -171,7 +171,7 @@ def fit_gaussian_process(self, fix_scale=False, verbose=False,
             np.abs(fluxes) /
             np.sqrt(flux_errors**2 + (1e-2 * np.max(fluxes))**2)
         )
-        scale = fluxes[signal_to_noises.idxmax()]
+        scale = np.abs(fluxes[signal_to_noises.idxmax()])
 
         kernel = (
             (0.2 * scale)**2 *

diff --git a/avocado/augment.py b/avocado/augment.py
@@ -223,6 +223,8 @@ def _choose_sampling_times(self, reference_object, augmented_metadata,
         )
         sampling_times = sampling_times.drop(drop_indices).copy()
 
+        sampling_times.reset_index(inplace=True, drop=True)
+
         return sampling_times
 
     def _simulate_light_curve_uncertainties(self, observations,