[ci] pin rl+hf tests to partial_dtensor (#4228)