From 4af32a1b3c525f13c795b38da286f75ec975f604 Mon Sep 17 00:00:00 2001
From: Ankit Goyal <angoyal@linkedin.com>
Date: Fri, 26 Jun 2020 13:00:41 -0700
Subject: [PATCH] Fix retries if the failure is due to missed heartbeats.
 (#451)

* Fix retries if the failures is due to missed heartbeats.

* upgrade release version to 0.3.28
---
 build.gradle                                                   | 2 +-
 .../src/main/java/com/linkedin/tony/ApplicationMaster.java     | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/build.gradle b/build.gradle
index 24b2c1dd..95cce0f6 100644
--- a/build.gradle
+++ b/build.gradle
@@ -92,7 +92,7 @@ ext.deps = [
 
 allprojects {
   group = "com.linkedin.tony"
-  project.version = "0.3.27"
+  project.version = "0.3.28"
 }
 
 task sourcesJar(type: Jar) {
diff --git a/tony-core/src/main/java/com/linkedin/tony/ApplicationMaster.java b/tony-core/src/main/java/com/linkedin/tony/ApplicationMaster.java
index c6a80ed4..77ff70f9 100644
--- a/tony-core/src/main/java/com/linkedin/tony/ApplicationMaster.java
+++ b/tony-core/src/main/java/com/linkedin/tony/ApplicationMaster.java
@@ -558,8 +558,9 @@ private void reset() {
                + container.getNodeId().getHost());
     }
 
-    // Reset the flag to track untracked processes.
+    // Reset the flags that indicate failure.
     untrackedTaskFailed = false;
+    taskHasMissesHB = false;
 
     // Reset session
     session = sessionBuilder.build();