🐺 🩹 Fix tensorboard tracker (#476)

src/pykeen/trackers/tensorboard.py

@@ -4,7 +4,7 @@
 
 import pathlib
 import time
-from typing import Any, Dict, Mapping, Optional, TYPE_CHECKING, Union
+from typing import Any, Mapping, Optional, TYPE_CHECKING, Union
 
 from .base import ResultTracker
 from ..constants import PYKEEN_LOGS

tests/test_trackers.py

@@ -2,11 +2,18 @@
 
 """Tests for result trackers."""
 
+import unittest
+
 from pykeen.trackers import TensorBoardResultTracker
 from pykeen.trackers.base import ConsoleResultTracker
 from pykeen.trackers.file import CSVResultTracker, JSONResultTracker
 from tests import cases
 
+try:
+    import tensorboard
+except ImportError:
+    tensorboard = None
+
 
 class CSVResultTrackerTests(cases.FileResultTrackerTests):
     """Tests for CSVResultTracker."""
@@ -26,7 +33,8 @@ class ConsoleResultTrackerTests(cases.ResultTrackerTests):
     cls = ConsoleResultTracker
 
 
+@unittest.skipIf(tensorboard is None, reason='TensorBoard is not installed')
 class TensorboardTrackerTests(cases.ResultTrackerTests):
-    """Tests for console tracker."""
+    """Tests for TensorBoard tracker."""
 
     cls = TensorBoardResultTracker

tox.ini

@@ -29,6 +29,7 @@ passenv =
     HOME
 extras =
     mlflow
+    tensorboard
     tests
 whitelist_externals =
     /bin/cat