fix individual plots

Author: ntellis

src/adam_impact_study/analysis/plots.py

@@ -619,27 +619,35 @@ def plot_individual_orbit_ip_over_time(
             logger.warning(f"No complete results found for orbit {orbit_id}")
             continue
 
-        # Sort by observation end time
-        mjd_times = ips.observation_end.mjd().to_numpy(zero_copy_only=False)
-        probabilities = ips.impact_probability.to_numpy(zero_copy_only=False)
-        sort_indices = mjd_times.argsort()
-        mjd_times = mjd_times[sort_indices]
-        probabilities = probabilities[sort_indices]
-
         # Plot sorted data on primary axis (MJD)
         ax1.set_xlabel("MJD")
         ax1.set_ylabel("Impact Probability")
         for condition in ips.condition_id.unique():
             results_at_condition = ips.select("condition_id", condition)
+
+            # Sort by observation end time
+            mjd_times = results_at_condition.observation_end.mjd().to_numpy(
+                zero_copy_only=False
+            )
+            print("proably unsorted?", mjd_times)
+            probabilities = results_at_condition.impact_probability.to_numpy(
+                zero_copy_only=False
+            )
+            sort_indices = mjd_times.argsort()
+            print("sort_indices", sort_indices)
+            mjd_times = mjd_times[sort_indices]
+            print("mjd_times_probably_sorted", mjd_times)
+            probabilities = probabilities[sort_indices]
+
             ax1.plot(
-                results_at_condition.observation_end.mjd(),
-                results_at_condition.impact_probability,
+                mjd_times,
+                probabilities,
                 label=condition,
                 lw=1,
             )
             ax1.scatter(
-                results_at_condition.observation_end.mjd(),
-                results_at_condition.impact_probability,
+                mjd_times,
+                probabilities,
                 label=condition,
             )