File size: 869 Bytes
714cf46 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | """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",
]
|