|
a |
|
b/sybil/__init__.py |
|
|
1 |
# type: ignore |
|
|
2 |
|
|
|
3 |
import sys |
|
|
4 |
|
|
|
5 |
if sys.version_info[:2] >= (3, 8): |
|
|
6 |
# TODO: Import directly (no need for conditional) when `python_requires = >= 3.8` |
|
|
7 |
from importlib.metadata import PackageNotFoundError, version # pragma: no cover |
|
|
8 |
else: |
|
|
9 |
from importlib_metadata import PackageNotFoundError, version # pragma: no cover |
|
|
10 |
|
|
|
11 |
try: |
|
|
12 |
# Change here if project is renamed and does not equal the package name |
|
|
13 |
dist_name = __name__ |
|
|
14 |
__version__ = version(dist_name) |
|
|
15 |
except PackageNotFoundError: # pragma: no cover |
|
|
16 |
__version__ = "unknown" |
|
|
17 |
finally: |
|
|
18 |
del version, PackageNotFoundError |
|
|
19 |
|
|
|
20 |
from sybil.model import Sybil |
|
|
21 |
from sybil.serie import Serie |
|
|
22 |
from sybil.utils.visualization import visualize_attentions, collate_attentions |
|
|
23 |
import sybil.utils.logging_utils |
|
|
24 |
|
|
|
25 |
__all__ = ["Sybil", "Serie", "visualize_attentions", "collate_attentions", "__version__"] |