From 4a6bc31f028211618cdd8f167589cc7cefb5689d Mon Sep 17 00:00:00 2001
From: hr2904 <hritik.rai@atlan.com>
Date: Fri, 23 Aug 2024 10:50:23 +0530
Subject: [PATCH] Added another exception catch

---
 .../atlas/repository/store/graph/v2/EntityGraphMapper.java | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphMapper.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphMapper.java
index 2cec8ff89c..8ae0b7bc14 100644
--- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphMapper.java
+++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphMapper.java
@@ -3091,7 +3091,12 @@ public void cleanUpClassificationPropagation(String classificationName) throws A
         }
         // Fetch all classificationVertex by classificationName and delete them if remaining
         for (AtlasVertex classificationVertex : classificationVertices) {
-            deleteDelegate.getHandler().deleteClassificationVertex(classificationVertex, true);
+            try {
+                deleteDelegate.getHandler().deleteClassificationVertex(classificationVertex, true);
+            }
+            catch (IllegalStateException e){
+                e.printStackTrace();
+            }
         }
         transactionInterceptHelper.intercept();
         LOG.info("Completed cleaning up classification {}", classificationName);