Improve optimistic update of datasetInfo (#398)

to reflect all fields from the config.
ServiceNow · Jan 27, 2023 · b569ca6 · b569ca6
1 parent 30e4c6d
commit b569ca6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/webapp/src/services/api.ts b/webapp/src/services/api.ts
@@ -327,11 +327,20 @@ export const api = createApi({
         );
         const patchDatasetInfo = dispatch(
           api.util.updateQueryData("getDatasetInfo", { jobId }, (draft) => {
+            if (partialConfig.name !== undefined) {
+              draft.projectName = partialConfig.name;
+            }
+            if (partialConfig.model_contract !== undefined) {
+              draft.modelContract = partialConfig.model_contract;
+            }
             if ("behavioral_testing" in partialConfig) {
               draft.perturbationTestingAvailable = Boolean(
                 partialConfig.behavioral_testing
               );
             }
+            if ("pipelines" in partialConfig) {
+              draft.predictionAvailable = Boolean(partialConfig.pipelines);
+            }
             if ("similarity" in partialConfig) {
               draft.similarityAvailable = Boolean(partialConfig.similarity);
             }