changes

Author: abidlabs

trackio/frontend/src/components/LinePlot.svelte

@@ -517,7 +517,7 @@
           aria-label="Reset horizontal zoom: show the full range on the x-axis"
         >
           <svg width="18" height="18" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2.25" stroke-linecap="round" stroke-linejoin="round">
-            <path d="M4 12h16M4 12l3-3M4 12l3 3M20 12l-3-3M20 12l-3 3"/>
+            <circle cx="11" cy="11" r="7"/><line x1="21" y1="21" x2="16.65" y2="16.65"/><line x1="8" y1="11" x2="14" y2="11"/>
           </svg>
         </button>
       {/if}
@@ -596,7 +596,7 @@
             aria-label="Reset horizontal zoom: show the full range on the x-axis"
           >
             <svg width="18" height="18" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2.25" stroke-linecap="round" stroke-linejoin="round">
-              <path d="M4 12h16M4 12l3-3M4 12l3 3M20 12l-3-3M20 12l-3 3"/>
+              <circle cx="11" cy="11" r="7"/><line x1="21" y1="21" x2="16.65" y2="16.65"/><line x1="8" y1="11" x2="14" y2="11"/>
             </svg>
           </button>
         {/if}

trackio/frontend/src/lib/dataProcessing.js

@@ -121,14 +121,24 @@ export function computeMetricPlotData(masterData, xColumn, metric, xLim) {
     (r) => r[metric] != null && r[metric] !== undefined,
   );
   if (xLim) {
-    const sorted = relevant.sort((a, b) => a[xColumn] - b[xColumn]);
-    let lo = 0;
-    let hi = sorted.length - 1;
-    while (lo < sorted.length && sorted[lo][xColumn] < xLim[0]) lo++;
-    while (hi >= 0 && sorted[hi][xColumn] > xLim[1]) hi--;
-    lo = Math.max(0, lo - 1);
-    hi = Math.min(sorted.length - 1, hi + 1);
-    relevant = sorted.slice(lo, hi + 1);
+    const groups = new Map();
+    for (const r of relevant) {
+      const key = `${r.run || ""}\0${r.data_type || "original"}`;
+      if (!groups.has(key)) groups.set(key, []);
+      groups.get(key).push(r);
+    }
+    const filtered = [];
+    for (const [, rows] of groups) {
+      rows.sort((a, b) => a[xColumn] - b[xColumn]);
+      let lo = 0;
+      let hi = rows.length - 1;
+      while (lo < rows.length && rows[lo][xColumn] < xLim[0]) lo++;
+      while (hi >= 0 && rows[hi][xColumn] > xLim[1]) hi--;
+      lo = Math.max(0, lo - 1);
+      hi = Math.min(rows.length - 1, hi + 1);
+      filtered.push(...rows.slice(lo, hi + 1));
+    }
+    relevant = filtered;
   }
   const originals = relevant.filter(
     (r) => r.data_type === "original" || !r.data_type,

trackio/frontend/src/pages/Metrics.svelte

@@ -140,13 +140,13 @@
     const countPerRunMetric = new Map();
     for (const r of originals) {
       const run = r.run;
-      for (const col of metricColumns) {
+      for (const col of cols) {
         if (r[col] == null) continue;
         const key = `${col}\0${run}`;
         countPerRunMetric.set(key, (countPerRunMetric.get(key) || 0) + 1);
       }
     }
-    const sp = new Set(metricColumns);
+    const sp = new Set(cols);
     for (const [key, count] of countPerRunMetric) {
       if (count > 1) {
         sp.delete(key.split("\0")[0]);