def before_training(self, strategy, **kwargs):
        """
        Check if the model is an instance of base class to ensure get_features()
        is implemented
        """
        if not isinstance(strategy.model, BaseModel):
            raise NotImplementedError(BaseModel.__name__ + ".get_features()")

        Check if the model is an instance of base class to ensure get_features()
        is implemented
        
avalanche/training/plugins/lfl.py