Think I worked it out. After the first record was added, it won't let me change the PartitionKey, but I can add a new one with the PartitionKey in root as shown!
I did look for information in the documentation, there was examples of geojson and partitions, just not together