Diaz went 3-for-4 with a homer for High-A Bradenton on Thursday as part of his rehab stint.
He's batting .391/.464/.522 in his first 28 plate appearances as he comes back from elbow surgery. Diaz could factor into the second half of Pittsburgh's catching situation if he can prove his elbow is healthy.