fe799dada87ef117262b19ee6776c359dfa6fb04,smac/runhistory/runhistory2epm.py,AbstractRunHistory2EPM,transform,#AbstractRunHistory2EPM#Any#,147

Before Change

        if self.impute_censored_data:
            // Get all censored runs
            c_run_dict = self.__select_runs(rh_data=copy.deepcopy(runhistory.data),
                                            select=RunType.CENSORED)
            if len(c_run_dict) == 0:
                self.logger.debug("No censored data found, skip imputation")

After Change

        if self.impute_censored_data:
            // Get all censored runs
            c_run_dict = {run: runhistory.data[run] for run in runhistory.data.keys()
                          if runhistory.data[run].status in self.impute_state and
                          runhistory.data[run].time < self.cutoff_time}
            if len(c_run_dict) == 0:
                self.logger.debug("No censored data found, skip imputation")

Instances