Merge pull request #386 from UCL-CCS/grid_search

Grid search

Author: wedeling

easyvvuq/actions/execute_local.py

@@ -268,7 +268,7 @@ def start(self, previous=None):
         for action in self.actions:
             previous = self.wrapper(action, previous)
         self.result = previous
-        assert(self.result['run_id'] == run_id)
+        assert (self.result['run_id'] == run_id)
         return previous
 
     def finished(self):

easyvvuq/analysis/pce_analysis.py

@@ -350,8 +350,8 @@ def sobols(P, coefficients):
                 varied = [_ for _ in self.sampler.vary.get_keys()]
                 S1 = {_: np.zeros(sobol.shape[-1]) for _ in varied}
                 ST = {_: np.zeros(sobol.shape[-1]) for _ in varied}
-                #S2 = {_ : {__: np.zeros(sobol.shape[-1]) for __ in varied} for _ in varied}
-                #for v in varied: del S2[v][v]
+                # S2 = {_ : {__: np.zeros(sobol.shape[-1]) for __ in varied} for _ in varied}
+                # for v in varied: del S2[v][v]
                 S2 = {_: np.zeros((len(varied), sobol.shape[-1])) for _ in varied}
                 for n, si in enumerate(sobol_idx):
                     if len(si) == 1:
@@ -360,8 +360,8 @@ def sobols(P, coefficients):
                     elif len(si) == 2:
                         v1 = varied[si[0]]
                         v2 = varied[si[1]]
-                        #S2[v1][v2] = sobol[n]
-                        #S2[v2][v1] = sobol[n]
+                        # S2[v1][v2] = sobol[n]
+                        # S2[v2][v1] = sobol[n]
                         S2[v1][si[1]] = sobol[n]
                         S2[v2][si[0]] = sobol[n]
                     for i in si: