Merge pull request #107 from sreeshnap/training/updatemodel

Training/updatemodel
singnet · Oct 6, 2022 · df9cba5 · df9cba5
2 parents bb07122 + 7044da4
commit df9cba5
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/packages/core/src/BaseServiceClient.js b/packages/core/src/BaseServiceClient.js
@@ -252,7 +252,7 @@ class BaseServiceClient {
     const request = await this._trainingUpdateModel(params);
     return new Promise((resolve, reject) => {
       this._modelServiceClient.update_model_access(request, (err, response) => {
-        logger.debug(`delete model ${err} ${response}`);
+        logger.debug(`update model ${err} ${response}`);
         if (err) {
           reject(err);
         } else {
@@ -265,7 +265,7 @@ class BaseServiceClient {
   async _trainingUpdateModel(params) {
     const message = "__update_model";
     const { currentBlockNumber, signatureBytes } =
-      await this._requestSignForModel(address, message);
+      await this._requestSignForModel(params.address, message);
 
     const ModelStateRequest = this._getUpdateModelRequestMethodDescriptor();
     const modelStateRequest = new ModelStateRequest();
@@ -283,11 +283,14 @@ class BaseServiceClient {
     modelDetailsRequest.setModelId(params.modelId);
     modelDetailsRequest.setGrpcMethodName(params.method);
     modelDetailsRequest.setGrpcServiceName(params.name);
-
+    modelDetailsRequest.setModelName(params.modelName);
     modelDetailsRequest.setDescription(params.description);
     modelDetailsRequest.setAddressListList(params.addressList);
     modelDetailsRequest.setTrainingDataLink("");
-    // modelDetailsRequest.setIsDefaultModel("");
+    modelDetailsRequest.setStatus(params.status);
+    modelDetailsRequest.setUpdatedDate(params.updatedDate);
+    modelDetailsRequest.setIsPubliclyAccessible(params.publicAccess);
+
 
     const { orgId, serviceId, groupId } = this.getServiceDetails();
     modelDetailsRequest.setOrganizationId(orgId);