def get_prediction()

in backend/time-series-forecasting/services/forecast_job_coordinator.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

    def get_prediction(self, job_id: str) -> Optional[pd.DataFrame]:
        """Get the prediction dataframe for a given job_id.

        Args:
            job_id (str): Job id.

        Returns:
            Optional[pd.DataFrame]: The prediction dataframe.
        """
        job = self._completed_jobs.get(job_id)

        if job is None:
            return None

        table_id = job.prediction_uri
        return self._get_bigquery_table_as_df(table_id=table_id) if table_id else None