diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java index e0eddfe4..bbd2474c 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java @@ -518,7 +518,7 @@ public class ExperimentServiceImpl implements ExperimentService { if (datasets != null) { for (Map dataset : datasets) { DatasetTempStorage datasetTempStorage = new DatasetTempStorage(); - datasetTempStorage.setRepoId(Integer.valueOf((String) dataset.get("dataset_id"))); + datasetTempStorage.setRepoId((Integer) dataset.get("dataset_id")); datasetTempStorage.setName((String) dataset.get("dataset_identifier")); datasetTempStorage.setVersion((String) dataset.get("dataset_version")); datasetTempStorage.setCreateBy((String) dataset.get("owner")); @@ -599,7 +599,7 @@ public class ExperimentServiceImpl implements ExperimentService { for (Map dataset : trainDatasetList) { if (!dataset.containsKey("dataset_ref")) { NewDatasetVo newDatasetVo = new NewDatasetVo(); - newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); + newDatasetVo.setId((Integer) dataset.get("dataset_id")); newDatasetVo.setName((String) dataset.get("dataset_name")); newDatasetVo.setVersion((String) dataset.get("dataset_version")); newDatasetVo.setIdentifier((String) dataset.get("dataset_identifier")); @@ -623,7 +623,7 @@ public class ExperimentServiceImpl implements ExperimentService { List> modelsList = (List>) modelTrainMap.get("models"); if (modelsList != null) { Map parentModelMap = modelsList.get(0); - String id = (String) parentModelMap.get("model_id"); + Integer id = (Integer) parentModelMap.get("model_id"); String identifier = (String) parentModelMap.get("model_identifier"); String version = (String) parentModelMap.get("model_version"); @@ -649,7 +649,7 @@ public class ExperimentServiceImpl implements ExperimentService { for (Map dataset : testDatasetList) { if (!dataset.containsKey("dataset_ref")) { NewDatasetVo newDatasetVo = new NewDatasetVo(); - newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); + newDatasetVo.setId((Integer) dataset.get("dataset_id")); newDatasetVo.setName((String) dataset.get("dataset_name")); newDatasetVo.setVersion((String) dataset.get("dataset_version")); newDatasetVo.setIdentifier((String) dataset.get("dataset_identifier")); @@ -672,7 +672,7 @@ public class ExperimentServiceImpl implements ExperimentService { for (int j = 0; j < modelTargetList.size(); j++) { Map targetModel = modelTargetList.get(i); - modelMetaVo.setId(Integer.valueOf((String) targetModel.get("model_id"))); + modelMetaVo.setId((Integer) targetModel.get("model_id")); modelMetaVo.setIdentifier((String) targetModel.get("model_identifier")); modelMetaVo.setName((String) targetModel.get("model_name")); modelMetaVo.setVersion((String) targetModel.get("model_version")); @@ -692,7 +692,7 @@ public class ExperimentServiceImpl implements ExperimentService { // } // modelMetaVo.setModelVersionVos(versionVos); - modelDependency.setRepoId(Integer.valueOf((String) targetModel.get("model_id"))); + modelDependency.setRepoId((Integer) targetModel.get("model_id")); modelDependency.setIdentifier((String) targetModel.get("model_identifier")); modelDependency.setVersion((String) targetModel.get("model_version")); modelDependency.setModelName((String) targetModel.get("model_name")); diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/WorkflowServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/WorkflowServiceImpl.java index eeae51cc..57d95ab8 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/WorkflowServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/WorkflowServiceImpl.java @@ -389,8 +389,12 @@ public class WorkflowServiceImpl implements WorkflowService { public void saveModelWorkFlow(Map inParameters, Long workflowId, String workflowName) throws Exception { try { Map model = (Map) inParameters.get("--model_name"); - LinkedHashMap value = (LinkedHashMap) model.get("value"); - insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((Integer) value.get("id")), (String) value.get("version"), Constant.Asset_Type_Model); + Object value = model.get("value"); + if (value instanceof String) { + return; + } + LinkedHashMap valueMap = (LinkedHashMap) model.get("value"); + insertAssetWorkFlow(workflowId, workflowName, Long.valueOf((Integer) valueMap.get("id")), (String) valueMap.get("version"), Constant.Asset_Type_Model); } catch (Exception e) { throw new Exception("模型参数有误:" + e); }