m

smithy-lang · Oct 8, 2024 · adb8ea8 · adb8ea8
1 parent d52c026
commit adb8ea8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/codegen/smithy-dafny-codegen/src/main/java/software/amazon/polymorph/utils/ModelUtils.java b/codegen/smithy-dafny-codegen/src/main/java/software/amazon/polymorph/utils/ModelUtils.java
@@ -735,12 +735,14 @@ public static List<Shape> getTopologicallyOrderedOrphanedShapesForService(
     TopologicalIndex topologicalIndex = TopologicalIndex.of(model);
 
     for (Shape shape : topologicalIndex.getOrderedShapes()) {
-      if (orphanedShapes.contains(shape)) {
+      if (orphanedShapes.contains(shape)
+      && shape.getId().getNamespace().equals(serviceShape.getId().getNamespace())) {
         orderedShapes.add(shape);
       }
     }
     for (Shape shape : topologicalIndex.getRecursiveShapes()) {
-      if (orphanedShapes.contains(shape)) {
+      if (orphanedShapes.contains(shape)
+        && shape.getId().getNamespace().equals(serviceShape.getId().getNamespace())) {
         orderedShapes.add(shape);
       }
     }