fix(aim): make an exception for checkSameSourceUrn for AIM case (#484)

Co-authored-by: Justin Donn <[email protected]>
linkedin · Dec 9, 2024 · ff330d3 · ff330d3
1 parent ae21b69
commit ff330d3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dao-api/src/main/java/com/linkedin/metadata/dao/utils/GraphUtils.java b/dao-api/src/main/java/com/linkedin/metadata/dao/utils/GraphUtils.java
@@ -23,6 +23,12 @@ public static void checkSameSourceUrn(@Nonnull final List<? extends RecordTempla
     if (relationships.isEmpty()) {
       return;
     }
+
+    // TODO: remove this temporary allow list after AIM no longer relies on ProducesLocalRelationshipBuilderFromFeatureDependencies
+    if (assetUrn != null && assetUrn.getEntityType().equals("mlFeatureVersion") && relationships.get(0).getClass().getSimpleName().equals("produces")) {
+      return;
+    }
+
     for (RecordTemplate relationship : relationships) {
       if (ModelUtils.isRelationshipInV2(relationship.schema())) {
         if (assetUrn == null) {