Merge pull request #449 from alfredliu-TRI/master

hotfix
TRI-AMDD · Aug 13, 2021 · f2088a2 · f2088a2
2 parents 4175688 + 6bdbd00
commit f2088a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/camd/agent/generic.py b/camd/agent/generic.py
@@ -80,7 +80,7 @@ def get_hypotheses(self, candidate_data, seed_data=None):
         t_pred, unc = self.pipeline.predict(self.candidate_data, return_std=True)
         t_pred += unc * self.alpha
         selected = np.argsort(-1.0 * t_pred)[: self.n_query]
-        return candidate_data.loc[selected]
+        return candidate_data.iloc[selected]
 
 
 class GPBatchUCB(HypothesisAgent):