Update scheduler.py

make SCHEDULER_QUEUE_CLASS works!