Fixed minor type checking issue.
@@ -22,12 +22,14 @@ from torch.nn import Module, Linear
from torch.utils.data import random_split, ConcatDataset, TensorDataset
from avalanche.benchmarks.scenarios import IStepInfo
+
from avalanche.training.strategies.strategy_callbacks import StrategyCallbacks
from avalanche.training.utils import copy_params_dict, zerolike_params_dict
if TYPE_CHECKING:
from avalanche.logging import StrategyLogger
from avalanche.evaluation import PluginMetric
+ from avalanche.training.strategies import BaseStrategy, JointTraining
PluggableStrategy = Union['BaseStrategy', 'JointTraining']
