The partnership does not report anything related to this "purchase" since it was you individually that purchased the units.
The partnership will file a final return through the date of sale. The income / loss will be allocated based on ownership up to the date of sale. Make sure you indicate that this is a final return and both K-1's are marked final.
All activity post sale transaction will be reported by you individually on your personal tax return on form Schedule C.
There are a number of issues here. I have attached a link to an IRS revenue ruling that explains what happens in this instance. It can be fairly complicated and depending on the $$ you may want to get some assistance from a tax professional.
https://www.irs.gov/pub/irs-drop/rr-99-6.pdf
*A reminder that posts in a forum such as this do not constitute tax advice.
Also keep in mind the date of replies, as tax law changes.
