Merge pull request #124 from x-tabdeveloping/setuptools-fix

Setuptools fix

Author: x-tabdeveloping

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 <power.up1163@gmail.com>", email = "martonkardos@cas.au.dk" }

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]):