Simplify arbitrary derivation

spotify · May 28, 2024 · dcadec9 · dcadec9
1 parent 8e79b32
commit dcadec9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scalacheck/src/main/scala-2/magnolify/scalacheck/ArbitraryDerivation.scala b/scalacheck/src/main/scala-2/magnolify/scalacheck/ArbitraryDerivation.scala
@@ -45,7 +45,7 @@ object ArbitraryDerivation {
               // pick a fixed subtype to have a chance to stop recursion
               Gen.const(subtypes.size + size)
             }
-          subtypeGen <- Gen.resize(size - 1, sealedTrait.subtypes(i).typeclass.arbitrary)
+          subtypeGen <- Gen.resize(size - 1, subtypes(i).typeclass.arbitrary)
         } yield subtypeGen
       }
     }

diff --git a/scalacheck/src/main/scala-3/magnolify/scalacheck/ArbitraryDerivation.scala b/scalacheck/src/main/scala-3/magnolify/scalacheck/ArbitraryDerivation.scala
@@ -45,7 +45,7 @@ object ArbitraryDerivation extends Derivation[Arbitrary]:
               // pick a fixed subtype to have a chance to stop recursion
               Gen.const(subtypes.size + size)
             }
-          subtypeGen <- Gen.resize(size - 1, sealedTrait.subtypes(i).typeclass.arbitrary)
+          subtypeGen <- Gen.resize(size - 1, subtypes(i).typeclass.arbitrary)
         } yield subtypeGen
       }
     }