add container update interface

bingduoduo1 · Jun 12, 2019 · e0f100c · e0f100c
1 parent 9082389
commit e0f100c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/...src/main/java/model/application/auto_train/base_interface/ArgumentContainerInterface.java b/...src/main/java/model/application/auto_train/base_interface/ArgumentContainerInterface.java
@@ -1,8 +1,11 @@
 package model.application.auto_train.base_interface;
 
+import model.config.GlobalException;
+
 public interface ArgumentContainerInterface {
     int getSize();
     boolean isValid();
     void saveObject2File();
     String getOutputFilePath();
+    void updateValue(String key, String value) throws GlobalException;
 }
diff --git a/...c/main/java/model/application/auto_train/pytorch_train/PytorchTrainArgumentContainer.java b/...c/main/java/model/application/auto_train/pytorch_train/PytorchTrainArgumentContainer.java
@@ -85,4 +85,17 @@ public void saveObject2File() {
         out.flush();
         out.close();
     }
+
+    @Override
+    public void updateValue(String key, String value) throws GlobalException {
+        if (mConfigMap.containsKey(key)) {
+            try {
+                mConfigMap.get(key).updateValue(value);
+            } catch (GlobalException e) {
+                throw e;
+            }
+        } else {
+            throw new GlobalException("invalid pytorch train argument");
+        }
+    }
 }