fdf356a91622033dc8e88ccf64c2f421e9d3a8a8,pycox/cox.py,CoxTime,predict_cumulative_hazard_at_times,#CoxTime#,833

Before Change

        return np.exp(-self.predict_cumulative_hazard(df, baseline_hazard_, batch_size, verbose))

    def predict_cumulative_hazard_at_times(self, times, df, batch_size=512, return_df=True):
        raise NotImplementedError("This isn"t as relevant when we have broken the propotionality.")

    def concordance_index(self, df, g_preds=None, batch_size=256):
        raise NotImplementedError()

After Change

            Pandas dataframe (or numpy matrix) [len(times), len(df)] with cumulative hazard
            estimates.
        """
        assert hasattr(self, "baseline_hazard_"), "Need to fit model first."
        if not hasattr(times, "__iter__"):
            times = [times]
        cum_haz = self.predict_cumulative_hazard(df, baseline_hazard_, batch_size, verbose)
        times_idx = search_sorted_idx(cum_haz.index.values, times)

Instances