Update YarnAllocator.scala