Fix empty datapoint in datapoint editor to ensure that CategoryLabel dropdown default values are reflected in base data.

nadah09 · LIT team · commit 960bf4e1645a · 2023-09-19T17:02:04.000-07:00
PiperOrigin-RevId: 566788498
diff --git a/lit_nlp/client/lib/types.ts b/lit_nlp/client/lib/types.ts
@@ -318,6 +318,10 @@ export function defaultValueByField(key: string, spec: Spec) {
   const fieldSpec: LitType = spec[key];
   // Explicitly check against undefined, as this value is often false-y if set.
   if (fieldSpec.default !== undefined) {
+    if (fieldSpec instanceof CategoryLabel &&
+        fieldSpec.vocab && !fieldSpec.vocab.includes(fieldSpec.default)) {
+      return fieldSpec.vocab[0];
+    }
     return fieldSpec.default;
   }
   // TODO(lit-dev): remove these and always use the spec default value.
