Fix gp tuner (#1592)

* fix gp tuner
microsoft · Oct 9, 2019 · 313b0f6 · 313b0f6
1 parent e93d2c2
commit 313b0f6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/azure-pipelines.yml b/azure-pipelines.yml
@@ -58,7 +58,7 @@ jobs:
   - script: |
       python3 -m pip install torch==0.4.1 --user
       python3 -m pip install torchvision==0.2.1 --user
-      python3 -m pip install tensorflow --user
+      python3 -m pip install tensorflow==1.13.1 --user
     displayName: 'Install dependencies for integration'
   - script: |
       source install.sh

diff --git a/src/sdk/pynni/nni/gp_tuner/gp_tuner.py b/src/sdk/pynni/nni/gp_tuner/gp_tuner.py
@@ -83,7 +83,7 @@ def update_search_space(self, search_space):
         """
         self._space = TargetSpace(search_space, self._random_state)
 
-    def generate_parameters(self, parameter_id):
+    def generate_parameters(self, parameter_id, **kwargs):
         """Generate next parameter for trial
         If the number of trial result is lower than cold start number,
         gp will first randomly generate some parameters.
@@ -123,7 +123,7 @@ def generate_parameters(self, parameter_id):
         logger.info("Generate paramageters:\n %s", results)
         return results
 
-    def receive_trial_result(self, parameter_id, parameters, value):
+    def receive_trial_result(self, parameter_id, parameters, value, **kwargs):
         """Tuner receive result from trial.
 
         Parameters