Account for error shapes as well

Author: RanVaknin

codegen/src/main/java/software/amazon/awssdk/codegen/AddShapes.java

@@ -36,6 +36,7 @@
 import software.amazon.awssdk.codegen.model.intermediate.ReturnTypeModel;
 import software.amazon.awssdk.codegen.model.intermediate.ShapeModel;
 import software.amazon.awssdk.codegen.model.intermediate.VariableModel;
+import software.amazon.awssdk.codegen.model.service.ErrorMap;
 import software.amazon.awssdk.codegen.model.service.Location;
 import software.amazon.awssdk.codegen.model.service.Member;
 import software.amazon.awssdk.codegen.model.service.Operation;
@@ -344,6 +345,16 @@ private boolean isDirectOperationInputOrOutput(Shape parentShape, Map<String, Sh
                     return true;
                 }
             }
+
+            if (operation.getErrors() != null) {
+                for (ErrorMap error : operation.getErrors()) {
+                    String errorShapeName = error.getShape();
+                    Shape outputShape = allC2jShapes.get(errorShapeName);
+                    if (parentShape.equals(outputShape)) {
+                        return true;
+                    }
+                }
+            }
         }
         return false;
     }