diff --git a/pyproject.toml b/pyproject.toml index 45add48..e3a9ff1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ profile = "black" [project] name = "turftopic" -version = "0.23.2" +version = "0.23.3" description = "Topic modeling with contextual representations from sentence transformers." authors = [ { name = "Márton Kardos ", email = "martonkardos@cas.au.dk" } diff --git a/turftopic/serialization.py b/turftopic/serialization.py index 81deb94..fa1ced1 100644 --- a/turftopic/serialization.py +++ b/turftopic/serialization.py @@ -1,10 +1,10 @@ import json import warnings +from importlib.metadata import version from pathlib import Path from typing import Union import joblib -import pkg_resources from huggingface_hub import snapshot_download IMPORTANT_PACKAGES = [ @@ -61,10 +61,7 @@ def get_package_versions() -> dict[str, str]: - return { - package: pkg_resources.get_distribution(package).version - for package in IMPORTANT_PACKAGES - } + return {package: version(package) for package in IMPORTANT_PACKAGES} def validate_package_versions(remote_versions: dict[str, str]):