Fix missing dependency in torch.utils.tensorboard (pytorch#115598)

Fixes pytorch#114591 Version package was removed in this pull request: pytorch#114108 but is still used in `torch.utils.tensorboard` causing import errors. The fix removes the import and uses a simpler check. Pull Request resolved: pytorch#115598 Approved by: https://github.com/malfet
dmenig · Dec 21, 2023 · dbd51f0 · dbd51f0
1 parent 35e9f5c
commit dbd51f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/utils/tensorboard/__init__.py b/torch/utils/tensorboard/__init__.py
@@ -1,5 +1,5 @@
 import tensorboard
-from packaging.version import Version
+from torch._vendor.packaging.version import Version
 
 if not hasattr(tensorboard, "__version__") or Version(
     tensorboard.__version__