retrieve right description from tags (G-Research#97)

jescalada · Jun 10, 2024 · b937ff7 · b937ff7
1 parent 71b8966
commit b937ff7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/src/services/models/runs/runDetailAppModel.ts b/src/src/services/models/runs/runDetailAppModel.ts
@@ -64,6 +64,7 @@ function getExperimentsData() {
 }
 
 function getRunInfo(runHash: string): IApiRequest<void> {
+  const DESCRIPTION_TAG = 'mlflow.note.content';
   if (getRunsInfoRequestRef) {
     getRunsInfoRequestRef.abort();
   }
@@ -74,6 +75,10 @@ function getRunInfo(runHash: string): IApiRequest<void> {
       const data = await getRunsInfoRequestRef.call((detail: any) => {
         exceptionHandler({ detail, model });
       });
+      data.props.description =
+        DESCRIPTION_TAG in data.params.tags
+          ? data.params.tags[DESCRIPTION_TAG]
+          : '';
       model.setState({
         runParams: data.params,
         runTraces: data.traces,
@@ -332,7 +337,7 @@ function editRunNameAndDescription(
             description,
           },
         });
-        if (res.id) {
+        if (res.ID) {
           onNotificationAdd({
             id: Date.now(),
             severity: 'success',