AutoTabPFNClassifier import fails with name 'TabPFNClassifierWrapper' is not defined

[nicoFhahn]

Description

A NameError occurs when attempting to import AutoTabPFNClassifier from tabpfn_extensions. The traceback indicates that TabPFNClassifierWrapper is referenced in get_tabpfn_models but has not been defined.

Environment

• OS: macOS
• Python Version: 3.12
• Environment: Virtual Environment (uv/.venv)

Installed Packages:

Package	Version
tabpfn	6.4.1
tabpfn-client	0.2.8
tabpfn-common-utils	0.2.16
tabpfn-extensions	0.2.2

Steps to Reproduce

Run the following import in the environment described above:

from tabpfn_extensions import AutoTabPFNClassifier

[priorphil]

Sorry for the late reply - I can't reproduce on my end in a freshly installed env, even with the same package versions. Would you mind pasting a full traceback?