Ended up figuring it out, I was using the system pyspark which was 4.0.0-preview and the proto was different so it wasn't parsing right.