savE (#719)

Co-authored-by: Piotr Mardziel <[email protected]>
truera · Dec 22, 2023 · 37772e2 · 37772e2
1 parent 426fa70
commit 37772e2
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 17 deletions.
diff --git a/trulens_eval/trulens_eval/pages/Evaluations.py b/trulens_eval/trulens_eval/pages/Evaluations.py
@@ -384,7 +384,7 @@ def highlight(s):
             match_query = None
 
             # Assumes record_json['perf']['start_time'] is always present
-            if val != record_json["perf"]["start_time"] and val != "":
+            if val != "":
                 match = None
                 for call in record.calls:
                     if call.perf.start_time.isoformat() == val:

diff --git a/...cord_viewer/dist/assets/index-9a8f0f59.js → ...cord_viewer/dist/assets/index-d2f862fd.js b/...cord_viewer/dist/assets/index-9a8f0f59.js → ...cord_viewer/dist/assets/index-d2f862fd.js
diff --git a/trulens_eval/trulens_eval/react_components/record_viewer/dist/index.html b/trulens_eval/trulens_eval/react_components/record_viewer/dist/index.html
@@ -4,7 +4,7 @@
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Record viewer</title>
-    <script type="module" crossorigin src="./assets/index-9a8f0f59.js"></script>
+    <script type="module" crossorigin src="./assets/index-d2f862fd.js"></script>
     <link rel="stylesheet" href="./assets/index-00b8e16d.css">
   </head>
   <body style="margin: 0">

diff --git a/trulens_eval/trulens_eval/react_components/record_viewer/src/RecordTable/RecordTableRow.tsx b/trulens_eval/trulens_eval/react_components/record_viewer/src/RecordTable/RecordTableRow.tsx
@@ -45,14 +45,18 @@ export default function RecordTableRowRecursive({
   const { startTime, timeTaken, endTime } = getStartAndEndTimesForNode(node);
 
   let selector = 'Select.App';
+
   if (node.path) selector += `.${node.path}`;
 
-  const isNodeSelected = selectedNode === node.raw?.perf.start_time;
+  const isRoot = !node.path;
+  const nodeStartTime = isRoot ? '' : node.raw?.perf.start_time;
+
+  const isNodeSelected = selectedNode === nodeStartTime;
 
   return (
     <>
       <TableRow
-        onClick={() => setSelectedNode(node.raw?.perf.start_time ?? undefined)}
+        onClick={() => setSelectedNode(nodeStartTime ?? undefined)}
         sx={{
           ...recordRowSx,
           background: isNodeSelected ? ({ palette }) => palette.primary.lighter : undefined,