presto-geospatial/src/main/java/io/prestosql/plugin/geospatial/SpatialPartitioningStateFactory.java/SpatialPartitioningStateFactory/setPartitionCount
class SpatialPartitioningStateFactory:        @Override
        public void setPartitionCount(int partitionCount)
        {
            this.partitionCounts.set(groupId, partitionCount);
        }