Merge pull request #9 from AutoResearch/renaming-to-convention

Fixed a bug that crashes the sampler
AutoResearch · Jul 4, 2023 · 7f2b5d5 · 7f2b5d5
2 parents 6a28861 + fdb5893
commit 7f2b5d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autora/experimentalist/sampler/model_disagreement/__init__.py b/src/autora/experimentalist/sampler/model_disagreement/__init__.py
@@ -63,6 +63,6 @@ def model_disagreement_sample(condition_pool: np.array, models: List, num_sample
     # sort the summed disagreements and select the top n
     idx = (-summed_disagreement).argsort()[:num_samples]
 
-    return X[idx]
+    return condition_pool[idx]
 
 model_disagreement_sampler = deprecated_alias(model_disagreement_sample, "model_disagreement_sampler")