Merge pull request #37 from MSDLLCpapers/bugfix

Bugfix - NParEGO default weights

obsidian/optimizer/bayesian.py

@@ -459,8 +459,8 @@ def _validate_hypers(self,
             if hps.get(key) is None:
                 if not defaults['optional']:
                     raise ValueError(f'Must specify hyperpameter value {key} for {aq_str}')
-                if key in ['weights']: #['scalarization_weights', 'weights']:
-                    aq_hps[key] = defaults['val'] * o_dim
+                if key in ['weights', 'scalarization_weights']:
+                    aq_hps[key] = [1] * o_dim
                 else:
                     aq_hps[key] = defaults['val']
 
@@ -553,7 +553,7 @@ def _parse_aq_kwargs(self,
 
         if aq == 'NParEGO':
             w = hps['scalarization_weights']
-            if isinstance(w,list):
+            if isinstance(w, list):
                 w = torch.tensor(w)
                 w = w/torch.sum(torch.abs(w))
             aq_kwargs['scalarization_weights'] = w