Merge pull request #2925 from atlanhq/exclude-classification-names

fix: add classificationNames not classifcations
atlanhq · Apr 9, 2024 · fe337fc · fe337fc
2 parents c0400c9 + bc9269e
commit fe337fc
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));