Fix pkg_resources deprecation warning

requirements.txt

@@ -3,6 +3,7 @@ numpy==1.24.4
 fsspec
 tensorboard
 packaging
+importlib-metadata;python_version<'3.8'
 psutil
 pyre_extensions
 typing_extensions

tests/utils/test_version.py

@@ -8,11 +8,16 @@
 # pyre-strict
 
 import unittest
-from unittest.mock import MagicMock, patch
+from unittest.mock import MagicMock, PropertyMock, patch
 
 import torch
 from packaging.version import Version
 
+try:
+    import importlib.metadata as importlib_metadata
+except ImportError:
+    import importlib_metadata
+
 from torchtnt.utils import version
 
 
@@ -47,6 +52,18 @@ def test_get_torch_version(self) -> None:
             self.assertNotEqual(version.get_torch_version(), Version("1.8.3"))
             self.assertEqual(version.get_torch_version(), Version("1.12.0"))
 
+    def test_get_torch_version_importlib(self) -> None:
+        with patch.object(torch, "__version__", new_callable=PropertyMock):
+            delattr(torch, "__version__")
+
+            with patch.object(importlib_metadata, "version", return_value="1.8.3"):
+                self.assertEqual(version.get_torch_version(), Version("1.8.3"))
+                self.assertNotEqual(version.get_torch_version(), Version("1.12.0"))
+
+            with patch.object(importlib_metadata, "version", return_value="1.12.0"):
+                self.assertNotEqual(version.get_torch_version(), Version("1.8.3"))
+                self.assertEqual(version.get_torch_version(), Version("1.12.0"))
+
     def test_torch_version_comparators(self) -> None:
         with patch.object(torch, "__version__", "2.0.0a0"):
             self.assertFalse(version.is_torch_version_geq("2.1.0"))

torchtnt/utils/version.py

@@ -9,10 +9,14 @@
 
 import platform
 
-import pkg_resources
 import torch
 from packaging.version import Version
 
+try:
+    import importlib.metadata as importlib_metadata
+except ImportError:
+    import importlib_metadata
+
 
 def is_windows() -> bool:
     """
@@ -48,8 +52,8 @@ def get_torch_version() -> Version:
         if hasattr(torch, "__version__"):
             pkg_version = Version(torch.__version__)
         else:
-            # try pkg_resources to infer version
-            pkg_version = Version(pkg_resources.get_distribution("torch").version)
+            # try importlib.metadata to infer version
+            pkg_version = Version(importlib_metadata.version("torch"))
     except TypeError as e:
         raise TypeError("PyTorch version could not be detected automatically.") from e