fix unnecessary import in torch.tensorboard. __init__.py

sachahu1 · Dec 11, 2023 · bf39212 · bf39212
1 parent 617c228
commit bf39212
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/torch/utils/tensorboard/__init__.py b/torch/utils/tensorboard/__init__.py
@@ -1,12 +1,9 @@
 import tensorboard
-from packaging.version import Version
 
-if not hasattr(tensorboard, "__version__") or Version(
-    tensorboard.__version__
-) < Version("1.15"):
+
+if not hasattr(tensorboard, "__version__") or (tensorboard.__version__ < "1.15"):
     raise ImportError("TensorBoard logging requires TensorBoard version 1.15 or above")
 
-del Version
 del tensorboard
 
 from .writer import FileWriter, SummaryWriter  # noqa: F401