Diarra has been out with an ankle problem since the middle of November.
He told The News: “I spoke to Lassana on Sunday after the West Brom game and he will be training by this weekend.
“If he doesn’t train Thursday then he might be training Friday – but he’ll definitely be back among the squad of players on Monday. So that’s great.
“Bolton might be too early – but I’m not going to rule him out.”