f754f90ed94c7db58f2d6573471b8a7d63828d87,test/test_automl/test_automl.py,AutoMLTest,test_delete_non_winning_models,#AutoMLTest#,118

Before 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([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))

After Change

        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])))
        ensemble_members_idx = set(automl.ensemble_.identifiers_)
        self.assertTrue(ensemble_members_idx.issubset(model_files_idx))

Instances