Add experience switch logging in WandBLogger [ADD TYPE_CHECKING]
""" This module handles all the functionalities related to the logging of
Avalanche experiments using Weights & Biases. """
-from typing import Union, List
+from typing import Union, List, TYPE_CHECKING
from pathlib import Path
import os
import errno
@@ -29,6 +29,10 @@ from avalanche.evaluation.metric_results import AlternativeValues, \
MetricValue, TensorImage
from avalanche.logging import StrategyLogger
+if TYPE_CHECKING:
+ from avalanche.evaluation.metric_results import MetricValue
+ from avalanche.training.strategies import BaseStrategy
+
class WandBLogger(StrategyLogger):
"""
