From a184896be00c537c38dbd2d8ead1235a23c3095a Mon Sep 17 00:00:00 2001
From: Elizabeth Mastrangelo <emmastrangelo@wpi.edu>
Date: Wed, 1 Apr 2026 08:05:50 -0400
Subject: [PATCH] Add complete TabPFNv2 usage comments in classification

Existing setup code fails when using TabPFNv2 because ModelVersion is never imported.
---
 capabilities/classification.mdx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/capabilities/classification.mdx b/capabilities/classification.mdx
index b251d89..77635a2 100644
--- a/capabilities/classification.mdx
+++ b/capabilities/classification.mdx
@@ -14,6 +14,8 @@ from tabpfn import TabPFNClassifier
 from sklearn.datasets import load_breast_cancer
 from sklearn.model_selection import train_test_split
 from sklearn.metrics import accuracy_score, log_loss
+# To use TabPFNv2:
+# from tabpfn.constants import ModelVersion
 
 X, y = load_breast_cancer(return_X_y=True)
 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
@@ -44,4 +46,4 @@ print("Log-Loss:", log_loss(y_test, probs))
   <Card title="Metric Tuning" icon="puzzle" href="/capabilities/metric-tuning">
     Tune scores other than maximum probability prediction such as F1 or balanced accuracy.
   </Card>
-</CardGroup>
\ No newline at end of file
+</CardGroup>