// Assert that the files of the models used by the ensemble weren"t deleted
model_files = backend_api.list_all_models(seed=seed)
model_files_idx = set([int(m_file.split(".")[-2]) for m_file in model_files])
ensemble_members_idx = set([idx[1] for idx in automl.ensemble_.identifiers_])
self.assertTrue(ensemble_members_idx.issubset(model_files_idx))
del automl
After Change
// Assert that the files of the models used by the ensemble weren"t deleted
model_files = backend_api.list_all_models(seed=seed)
model_files_idx = set()
for m_file in model_files:
// Extract the model identifiers from the filename
m_file = os.path.split(m_file)[1].replace(".model", "").split(".", 2)
model_files_idx.add((int(m_file[0]), int(m_file[1]), float(m_file[2])))