From f42f25021cd6c0c38bfe38d995d289a7800257ed Mon Sep 17 00:00:00 2001 From: Akash Verma Date: Wed, 30 Oct 2024 15:41:00 -0700 Subject: [PATCH] Use importlib metadata version since setuptools pkg_resources is deprecated. --- openhtf/__init__.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/openhtf/__init__.py b/openhtf/__init__.py index 0dd2e3672..ce6a88822 100644 --- a/openhtf/__init__.py +++ b/openhtf/__init__.py @@ -13,6 +13,7 @@ # limitations under the License. """The main OpenHTF entry point.""" +import importlib.metadata import signal import typing @@ -36,7 +37,6 @@ from openhtf.util import functions from openhtf.util import logs from openhtf.util import units -import pkg_resources __all__ = ( # Expliclty export certain API components. # Modules. @@ -133,14 +133,11 @@ def get_version(): - """Returns the version string of the 'openhtf' package. - - Note: the version number doesn't seem to get properly set when using ipython. - """ + """Returns the version string of the 'openhtf' package.""" try: - return pkg_resources.get_distribution('openhtf') + return importlib.metadata.version('openhtf') except pkg_resources.DistributionNotFound: - return 'Unknown - Perhaps openhtf was not installed via setup.py or pip.' + return 'Unknown - openhtf not installed via pip.' __version__ = get_version()