Expose package version

Author: lukasgarbas

transformer_ranker/__init__.py

@@ -1,5 +1,9 @@
+from importlib.metadata import version
+
 from .embedder import Embedder
 from .ranker import TransformerRanker
 from .utils import Result, prepare_popular_models
 
+__version__ = version("transformer-ranker")
+
 __all__ = ["TransformerRanker", "Result", "prepare_popular_models"]

transformer_ranker/datacleaner.py

@@ -13,7 +13,7 @@
 
 
 class TaskCategory(str, Enum):
-    """Supported task categories"""
+    """Supported tasks"""
 
     TEXT_CLASSIFICATION = "text classification"
     TEXT_PAIR_CLASSIFICATION = "text pair classification"