def get_category_rows()

in databricks/lib/repository/ground_truth/stats.py [0:0]

• 26 lines of code
• 7 McCabe index (conditional complexity)

    def get_category_rows(self, job_ids: List[int]) -> List[List[Any]]:

        stats = self.calculate_stats_by_category(job_ids)
        rows = []

        rows.extend(self.get_precision_rows(job_ids, include_files=False))
        for category in stats:
            row = ["- " + category]
            for job_id in job_ids:
                row.append(stats[category][job_id]["precision"])

            rows.append(row)

        rows.extend(self.get_recall_rows(job_ids, include_files=False))
        for category in stats:
            row = ["- " + category]
            for job_id in job_ids:
                row.append(stats[category][job_id]["recall"])

            rows.append(row)

        rows.extend(
            self.get_accuracy_rows(
                job_ids, include_files=False, include_categories=False
            )
        )
        for category in stats:
            row = ["- " + category]
            for job_id in job_ids:
                row.append(stats[category][job_id]["accuracy"])

            rows.append(row)

        return rows