DATA-3969 - Add custom args to training metadata (#693)

Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com>

Author: tahiyasalam

app/mltraining/v1/ml_training.pb.go

@@ -290,6 +290,8 @@ export class TrainingJobMetadata extends jspb.Message {
   setTagsList(value: Array<string>): void;
   addTags(value: string, index?: number): string;
 
+  getArgumentsMap(): jspb.Map<string, string>;
+  clearArgumentsMap(): void;
   serializeBinary(): Uint8Array;
   toObject(includeInstance?: boolean): TrainingJobMetadata.AsObject;
   static toObject(includeInstance: boolean, msg: TrainingJobMetadata): TrainingJobMetadata.AsObject;
@@ -320,6 +322,7 @@ export namespace TrainingJobMetadata {
     trainingEnded?: google_protobuf_timestamp_pb.Timestamp.AsObject,
     syncedModelId: string,
     tagsList: Array<string>,
+    argumentsMap: Array<[string, string]>,
   }
 }
 

gen/js/app/mltraining/v1/ml_training_pb.d.ts

@@ -1942,7 +1942,8 @@ proto.viam.app.mltraining.v1.TrainingJobMetadata.toObject = function(includeInst
     trainingStarted: (f = msg.getTrainingStarted()) && google_protobuf_timestamp_pb.Timestamp.toObject(includeInstance, f),
     trainingEnded: (f = msg.getTrainingEnded()) && google_protobuf_timestamp_pb.Timestamp.toObject(includeInstance, f),
     syncedModelId: jspb.Message.getFieldWithDefault(msg, 5, ""),
-    tagsList: (f = jspb.Message.getRepeatedField(msg, 16)) == null ? undefined : f
+    tagsList: (f = jspb.Message.getRepeatedField(msg, 16)) == null ? undefined : f,
+    argumentsMap: (f = msg.getArgumentsMap()) ? f.toObject(includeInstance, undefined) : []
   };
 
   if (includeInstance) {
@@ -2056,6 +2057,12 @@ proto.viam.app.mltraining.v1.TrainingJobMetadata.deserializeBinaryFromReader = f
       var value = /** @type {string} */ (reader.readString());
       msg.addTags(value);
       break;
+    case 21:
+      var value = msg.getArgumentsMap();
+      reader.readMessage(value, function(message, reader) {
+        jspb.Map.deserializeBinary(message, reader, jspb.BinaryReader.prototype.readString, jspb.BinaryReader.prototype.readString, null, "", "");
+         });
+      break;
     default:
       reader.skipField();
       break;
@@ -2216,6 +2223,10 @@ proto.viam.app.mltraining.v1.TrainingJobMetadata.serializeBinaryToWriter = funct
       f
     );
   }
+  f = message.getArgumentsMap(true);
+  if (f && f.getLength() > 0) {
+    f.serializeBinary(21, writer, jspb.BinaryWriter.prototype.writeString, jspb.BinaryWriter.prototype.writeString);
+  }
 };
 
 
@@ -2657,6 +2668,28 @@ proto.viam.app.mltraining.v1.TrainingJobMetadata.prototype.clearTagsList = funct
 };
 
 
+/**
+ * map<string, string> arguments = 21;
+ * @param {boolean=} opt_noLazyCreate Do not create the map if
+ * empty, instead returning `undefined`
+ * @return {!jspb.Map<string,string>}
+ */
+proto.viam.app.mltraining.v1.TrainingJobMetadata.prototype.getArgumentsMap = function(opt_noLazyCreate) {
+  return /** @type {!jspb.Map<string,string>} */ (
+      jspb.Message.getMapField(this, 21, opt_noLazyCreate,
+      null));
+};
+
+
+/**
+ * Clears values from the map. The map will be non-null.
+ * @return {!proto.viam.app.mltraining.v1.TrainingJobMetadata} returns this
+ */
+proto.viam.app.mltraining.v1.TrainingJobMetadata.prototype.clearArgumentsMap = function() {
+  this.getArgumentsMap().clear();
+  return this;};
+
+
 
 
 

gen/js/app/mltraining/v1/ml_training_pb.js

@@ -123,6 +123,7 @@ message TrainingJobMetadata {
   google.protobuf.Timestamp training_ended = 10 [(tagger.v1.tags) = "bson:\"training_ended\" json:\"training_ended\""];
   string synced_model_id = 5 [(tagger.v1.tags) = "bson:\"synced_model_id\" json:\"synced_model_id\""];
   repeated string tags = 16 [(tagger.v1.tags) = "bson:\"tags\" json:\"tags\""];
+  map<string, string> arguments = 21 [(tagger.v1.tags) = "bson:\"arguments\" json:\"arguments\""];
 
   reserved 1, 6;
   reserved "request", "user_email";