But to answer your question directly: the amendments to the withdrawal bill deliberately excluded the EU social rights. So no EU can be certain of what prisoners rights will be once we leave. They can guess, of course, but there is no legal assurance and that is what is needed. Hence an EU to UK extradition is legally challengeable.

In thw othwr direction the rights are known, so I doubt if a legal challenge would aucceed.