fix: add classificationNames not classifcations

atlanhq · Apr 9, 2024 · bc9269e · bc9269e
1 parent 67a1895
commit bc9269e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...sitory/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java b/...sitory/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java
@@ -1025,7 +1025,7 @@ private AtlasEntityHeader mapVertexToAtlasEntityHeader(AtlasVertex entityVertex,
             if(includeClassifications){
                 ret.setClassificationNames(getAllTraitNamesFromAttribute(entityVertex));
             } else if (!includeClassifications && includeClassificationNames) {
-                ret.setClassifications(getAllClassifications(entityVertex));
+                ret.setClassificationNames(getAllTraitNamesFromAttribute(entityVertex));
             }
             ret.setIsIncomplete(isIncomplete);
             ret.setLabels(getLabels(entityVertex));