[replay] fix: need to import BaseStrategy for type checking
import random
from abc import ABC, abstractmethod
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, TYPE_CHECKING
import torch
from numpy import inf
@@ -15,6 +15,9 @@ from avalanche.benchmarks.utils.data_loader import \
from avalanche.models import FeatureExtractorBackbone
from avalanche.training.plugins.strategy_plugin import StrategyPlugin
+if TYPE_CHECKING:
+ from avalanche.training.strategies import BaseStrategy
+
class ReplayPlugin(StrategyPlugin):
"""
