Fix up collection selector

soto-project · Oct 16, 2024 · 83db1af · 83db1af
1 parent 81d3400
commit 83db1af
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Sources/SotoSmithy/Selectors/BasicSelectors.swift b/Sources/SotoSmithy/Selectors/BasicSelectors.swift
@@ -64,7 +64,7 @@ public struct SimpleTypeSelector: Selector {
 public struct CollectionSelector: Selector {
     public init() {}
     public func select(using model: Model, shape: Shape) -> Bool {
-        return shape is CollectionShape
+        return shape is ListShape
     }
 }
 

diff --git a/Sources/SotoSmithyAWS/Traits/AwsCoreTraits.swift b/Sources/SotoSmithyAWS/Traits/AwsCoreTraits.swift
@@ -74,7 +74,7 @@ public struct AwsDataTrait: SingleValueTrait {
     public var selector: Selector {
         return OrSelector(
             SimpleTypeSelector(),
-            CollectionSelector(),
+            TypeSelector<ListShape>(),
             TypeSelector<StructureShape>(),
             TypeSelector<UnionShape>(),
             TypeSelector<MemberShape>()