Merge pull request #1295 from guardrails-ai/chore/remove-pkg-resources

Replace pkg_resources

Author: CalebCourier

guardrails/hub/install.py

@@ -3,7 +3,7 @@
 from string import Template
 from typing import Callable, cast, List
 
-import pkg_resources
+import importlib.metadata
 
 from guardrails.hub.validator_package_service import (
     ValidatorPackageService,
@@ -149,7 +149,7 @@ def install(
     installed_version_message = ""
     with contextlib.suppress(Exception):
         package_name = ValidatorPackageService.get_normalized_package_name(validator_id)
-        installed_version = pkg_resources.get_distribution(package_name).version
+        installed_version = importlib.metadata.version(package_name)
         if installed_version:
             installed_version_message = f" version {installed_version}"
 

tests/unit_tests/hub/test_hub_install.py

@@ -115,8 +115,8 @@ def test_install_local_models__true(self, mocker, use_remote_inferencing):
 
         mock_logger_log = mocker.patch("guardrails.hub.install.cli_logger.log")
 
-        pkg_resources = mocker.patch("guardrails.hub.install.pkg_resources")
-        pkg_resources.get_distribution.return_value.version = "1.0.0"
+        importlib_metadata = mocker.patch("guardrails.hub.install.importlib.metadata")
+        importlib_metadata.version = "1.0.0"
 
         get_manifest_and_site_packages_mock = mocker.patch(
             "guardrails.hub.validator_package_service.ValidatorPackageService.get_manifest_and_site_packages"