was not giving set to partition - fixed

- giving list of buckets to store and it'll cycle over it
- Some comments in scheduler.py
lint fixes