Update restricted_partitions.py