pytorch#69894: setuptools LooseVersion missing from the latest setupt…

…ools, so migrated to using packaging.version.Version.
keskival · Dec 14, 2021 · fd0207a · fd0207a
1 parent 51033ec
commit fd0207a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -3,6 +3,7 @@ astunparse
 expecttest
 future
 numpy
+packaging
 psutil
 pyyaml
 requests

diff --git a/torch/utils/tensorboard/__init__.py b/torch/utils/tensorboard/__init__.py
@@ -1,12 +1,9 @@
 import tensorboard
-from setuptools import distutils
-
-LooseVersion = distutils.version.LooseVersion
+from packaging.version import Version as LooseVersion
 
 if not hasattr(tensorboard, '__version__') or LooseVersion(tensorboard.__version__) < LooseVersion('1.15'):
     raise ImportError('TensorBoard logging requires TensorBoard version 1.15 or above')
 
-del distutils
 del LooseVersion
 del tensorboard