diff --git a/src/intcox.rs b/src/intcox.rs index 4aa11f4..6378780 100644 --- a/src/intcox.rs +++ b/src/intcox.rs @@ -191,8 +191,8 @@ pub fn fit_interval_censored_cox(data_times: Matrix2xX, mut data_indep: DMa // Get time points (t_0 = 0, t_1, ..., t_m) // TODO: Reimplement Turnbull intervals - let mut time_points: Vec; - time_points = data_times.iter().copied().collect(); + let mut time_points: Vec = Vec::with_capacity(data_times.len() + 1); + time_points.extend(data_times.iter()); time_points.push(0.0); // Ensure 0 is in the list //time_points.push(f64::INFINITY); // Ensure infinity is on the list time_points.sort_by(|a, b| a.partial_cmp(b).unwrap()); // Cannot use .sort() as f64 does not implement Ord