The better driveshaft shops wil want the front and rear yokes your are planning to use during the balancing.
Have you experience at changing ring and pinion gears,
Have you thoroughly checked that your trans was not damaged from the driveshaft failure, did you check your trans overrunning clutch for damage after the driveshaft failure. You could be flirting a trans explosion if the roller clutch was damaged from the driveshaft failure.