| """Probes package exports. | |
| This enables convenient imports like: | |
| from protify.probes import TransformerForSequenceClassification, LinearProbe | |
| Works both when the repository is the main module and when it is used as a | |
| submodule in another project (so long as `src/` is on PYTHONPATH or the | |
| package is installed). | |
| """ | |
| from .linear_probe import LinearProbe, LinearProbeConfig # noqa: F401 | |
| from .transformer_probe import ( | |
| TransformerForSequenceClassification, | |
| TransformerForTokenClassification, | |
| TransformerProbeConfig, | |
| ) # noqa: F401 | |
| from .packaged_probe_model import PackagedProbeConfig, PackagedProbeModel # noqa: F401 | |
| __all__ = [ | |
| "LinearProbe", | |
| "LinearProbeConfig", | |
| "TransformerForSequenceClassification", | |
| "TransformerForTokenClassification", | |
| "TransformerProbeConfig", | |
| "PackagedProbeConfig", | |
| "PackagedProbeModel", | |
| ] | |