From 29d5d4126014471dea3be8f8232ef10a554aeb19 Mon Sep 17 00:00:00 2001
From: Jonathan Eagles <jeagles@apache.org>
Date: Tue, 16 Apr 2024 12:21:59 -0500
Subject: [PATCH] TEZ-4553. Default task scheduler to DagAwareTaskScheduler to
 avoid hang in TEZ-3535

---
 .../src/main/java/org/apache/tez/dag/api/TezConfiguration.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tez-api/src/main/java/org/apache/tez/dag/api/TezConfiguration.java b/tez-api/src/main/java/org/apache/tez/dag/api/TezConfiguration.java
index 3dc6fe4745..7e86853631 100644
--- a/tez-api/src/main/java/org/apache/tez/dag/api/TezConfiguration.java
+++ b/tez-api/src/main/java/org/apache/tez/dag/api/TezConfiguration.java
@@ -905,7 +905,7 @@ public TezConfiguration(boolean loadDefaults) {
   @ConfigurationProperty
   public static final String TEZ_AM_YARN_SCHEDULER_CLASS = TEZ_AM_PREFIX + "yarn.scheduler.class";
   public static final String TEZ_AM_YARN_SCHEDULER_CLASS_DEFAULT =
-      "org.apache.tez.dag.app.rm.YarnTaskSchedulerService";
+      "org.apache.tez.dag.app.rm.DagAwareYarnTaskScheduler";
 
   /**
    * Int value. The AM waits this amount of time when the first DAG is submitted but not all the services are ready.