3ce15cf0b1b83503d0a35a0077cb93322c2cc710,finetune/base.py,BaseModel,_inference,#BaseModel#Any#Any#,239

Before Change

        predictions = [None] * n
        for i in range(n):
            y = next(self._predictions)
            y = y[mode] if mode else y
            predictions[i] = y
        return predictions
        

After Change

    def _inference(self, Xs, mode=None):
        Xs = self.input_pipeline._format_for_inference(Xs)
        if self._cached_predict:
            return self._cached_inference(Xs=Xs, mode=mode)
        else:
            estimator = self.get_estimator()
            input_fn = self.input_pipeline.get_predict_input_fn(Xs)
            length = len(Xs) if not callable(Xs) else None

            predictions = tqdm.tqdm(
                estimator.predict(
                    input_fn=input_fn, predict_keys=mode
                ),
                total=length,
                desc="Inference"
            )
            return [pred[mode] if mode else pred for pred in predictions]
        
    def fit(self, *args, **kwargs):
         An alias for finetune. 
        return self.finetune(*args, **kwargs)

Instances