22335: Exclude protected labels from generation output

howsoai · Nov 26, 2024 · ce2bd42 · ce2bd42
1 parent b54c53e
commit ce2bd42
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/codegen/engine.ts b/codegen/engine.ts
@@ -45,6 +45,7 @@ export interface LabelDefinition {
   description?: string | null;
   use_active_session?: boolean;
   attribute?: SchemaType | null;
+  protected?: boolean;
   payload?: boolean;
   long_running?: boolean;
   read_only?: boolean;

diff --git a/codegen/generator.ts b/codegen/generator.ts
@@ -39,14 +39,13 @@ export class Generator {
       "debug_label",
       "initialize",
       "initialize_for_deployment",
-      "set_contained_trainee_maps",
       "version",
       "get_api",
       "single_react",
       "single_react_series",
       ...Object.entries(doc.labels).reduce<string[]>((ignored, [label, def]) => {
-        // Ignore all the attribute labels
-        if (def.attribute != null) ignored.push(label);
+        // Ignore all the attribute and protected labels
+        if (def.attribute != null || def.protected) ignored.push(label);
         return ignored;
       }, []),
     ];