kgmodule-utils 0.2.2__tar.gz → 0.2.3__tar.gz

{kgmodule_utils-0.2.2 → kgmodule_utils-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kgmodule-utils
-Version: 0.2.2
+Version: 0.2.3
 Summary: Shared types and snapshot infrastructure for the KGModule SDK
 License: Elastic-2.0
 License-File: LICENSE

{kgmodule_utils-0.2.2 → kgmodule_utils-0.2.3}/pyproject.toml

@@ -10,7 +10,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "kgmodule-utils"
-version = "0.2.2"
+version = "0.2.3"
 description = "Shared types and snapshot infrastructure for the KGModule SDK"
 readme = "README.md"
 license = { text = "Elastic-2.0" }
@@ -52,8 +52,29 @@ target-version = ["py312", "py313"]
 line-length    = 100
 target-version = "py312"
 
+[tool.pylint.main]
+source-roots = ["src"]
+init-hook = "import sys; sys.path.insert(0, 'src')"
+
+[tool.pylint."messages control"]
+disable = [
+    "missing-module-docstring",
+]
+
 [tool.mypy]
 python_version     = "3.12"
 strict             = true
 warn_unused_ignores = true
 disallow_untyped_defs = true
+
+[[tool.mypy.overrides]]
+module = [
+    "sentence_transformers.*",
+    "transformers.*",
+    "numpy.*",
+]
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "kg_utils.embedder"
+disallow_untyped_calls = false

{kgmodule_utils-0.2.2 → kgmodule_utils-0.2.3}/src/kg_utils/__init__.py

@@ -9,4 +9,4 @@ Sub-packages:
                           wrap_embedder(), load_sentence_transformer().
 """
 
-__version__ = "0.2.2"
+__version__ = "0.2.3"

{kgmodule_utils-0.2.2 → kgmodule_utils-0.2.3}/src/kg_utils/embedder.py

@@ -95,6 +95,15 @@ def load_sentence_transformer(model_name: str = DEFAULT_MODEL) -> Any:
     """
     from sentence_transformers import SentenceTransformer  # pylint: disable=import-outside-toplevel
 
+    try:
+        from transformers import logging as hf_logging  # pylint: disable=import-outside-toplevel
+
+        hf_logging.set_verbosity_error()
+    except ImportError:
+        pass
+
+    os.environ["TQDM_DISABLE"] = "1"
+
     resolved = KNOWN_MODELS.get(model_name, model_name)
     trust_remote = "nomic-ai/" in resolved
     local_path = resolve_model_path(resolved)
@@ -133,7 +142,7 @@ class SentenceTransformerEmbedder(Embedder):
         try:
             from transformers import logging as hf_logging  # pylint: disable=import-outside-toplevel
 
-            hf_logging.set_verbosity_error()  # type: ignore[no-untyped-call]
+            hf_logging.set_verbosity_error()
         except ImportError:
             pass