Na, dich hat es aber schwer erwischt.
Vermutlich schweifen deine Gedanken insbesondere dann ab, wenn du alleine vor dich hinarbeitest. Versuche doch in dieser zeitlich befristeten Phase mehr mit anderen zusammenzuarbeiten, wenn das geht. Nehme mehr an Meetings teil, bei denen du durch Mitarbeiter abgelenkt wirst.
Ansonsten kannst du dem "Problem" auch ausweichen, indem du öfter mal früher nach Hause gehst oder ganz Urlaub machst, wenn das aktuell möglich ist. Vielleicht hat dein (neuer?) Partner ja auch gerade die Möglichkeit, jobmäßig kürzer zu treten.
Wenn du ein gutes Verhältnis zu deinen Kollegen hast, würde ich denen auch einfach mitteilen, wie es aktuell um dich steht. Die werden schon Verständnis dafür haben, dass du momentan nicht so produktiv bist.