Reduced block number of DictRDD in grid_search test