Set forgetting_threshold in test