Merge branch 'main' of https://github.com/MSDLLCpapers/obsidian

MSDLLCpapers · Aug 21, 2024 · f14dd3a · f14dd3a
2 parents 78a311e + f88ee78
commit f14dd3a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/obsidian/optimizer/bayesian.py b/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