updated so shuffle manager is only specified for mllib