Error when running random posthoc ensembling on tabrepo

[arberzela]

Things look fine on tabrepo except the random posthoc ensembler. When running the following:

PYTHONPATH=. python SearchingOptimalEnsembles_experiments/main.py --no_wandb --worker_dir SearchingOptimalEnsembles_experiments/  --project_name SOE_tabrepo --num_iterations 100 --metric_name error --metadataset_name tabrepo --dataset_id 8 --run_name random1_8_4 --meta_split_id 4 --searcher_name random --max_num_pipelines 5 --experiment_group random1_0 --ensembler_name random --data_version version3_class

I get the following error:

Traceback (most recent call last):
  File "/home/zelaa/SearchingOptimalEnsembles/SearchingOptimalEnsembles_experiments/main.py", line 130, in <module>
    SOE.run(**config)
  File "/home/zelaa/SearchingOptimalEnsembles/SearchingOptimalEnsembles/api.py", line 276, in run
    incumbent_ensemble, incumbent = posthoc_ensembler.sample(
  File "/home/zelaa/SearchingOptimalEnsembles/SearchingOptimalEnsembles/posthoc/random_ensembler.py", line 44, in sample
    ensembles = self.sampler.generate_ensembles(
  File "/home/zelaa/SearchingOptimalEnsembles/SearchingOptimalEnsembles/samplers/random_sampler.py", line 27, in generate_ensembles
    return candidates[ensembles].tolist()
TypeError: only integer scalar arrays can be converted to a scalar index