My understanding is that when particles are entangled with each other, they must be described together with one wave function rather than one wave function for

My understanding is that when particles are entangled with each other, they must be described together with one wave function rather than one wave function for each of them. So once you measure particle A, the wave function collapses both for A and B.