Add column for CVE knowledge

backend/app/services/leaderboard.py

@@ -28,6 +28,9 @@ class LeaderboardService:
             )["train"]
 
             df = dataset.to_pandas()
+
+            # Drop NaN values before converting to dict
+            df = df.replace({float('nan'): None})
             data = df.to_dict("records")
 
             stats = {
@@ -133,6 +136,11 @@ class LeaderboardService:
                     "value": data.get("secure_coding", 0),
                     "normalized_score": data.get("secure_coding", 0),
                 },
+                "cve_knowledge": {
+                    "name": "CVE Knowledge",
+                    "value": data.get("cve_knowledge", 0),
+                    "normalized_score": data.get("cve_knowledge", 0),
+                },
             }
 
             features = {

frontend/src/pages/LeaderboardPage/components/Leaderboard/constants/defaults.js

@@ -137,6 +137,12 @@ const COLUMNS = {
       defaultVisible: true,
       label: "Secure coding practices",
     },
+    "evaluations.cve_knowledge.value": {
+      group: "evaluation",
+      size: COLUMN_SIZES.BENCHMARK,
+      defaultVisible: true,
+      label: "CVE Knowledge",
+    },
   },
   MODEL_INFO: {
     "metadata.co2_cost": {

frontend/src/pages/LeaderboardPage/components/Leaderboard/utils/columnUtils.js

@@ -759,6 +759,15 @@ export const createColumns = (
         "evaluations.secure_coding.value"
       ],
     },
+    {
+      accessorKey: "evaluations.cve_knowledge.value",
+      header: createHeaderCell("CVE Knowledge"),
+      cell: ({ row, getValue }) =>
+        createScoreCell(getValue, row, "evaluations.cve_knowledge.value"),
+      size: TABLE_DEFAULTS.COLUMNS.COLUMN_SIZES[
+        "evaluations.cve_knowledge.value"
+      ],
+    },
     {
       accessorKey: "evaluations.safetensors.value",
       header: createHeaderCell("Safetensors Usage"),