Revert some changes for brivity

optuna · Dec 10, 2024 · a28e471 · a28e471
1 parent ef1481c
commit a28e471
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/package/samplers/hebo/sampler.py b/package/samplers/hebo/sampler.py
@@ -151,7 +151,7 @@ def _sample_relative_stateless(
         params = pd.DataFrame([t.params for t in trials])
         values[np.isnan(values)] = worst_value
         values *= sign
-        hebo.observe(params, values[:, np.newaxis])
+        hebo.observe(params, values)
         return self._suggest_and_transform_to_dict(hebo, search_space)
 
     def sample_relative(