Handle poly functions in functionArgsInfo

scala · Jan 13, 2023 · ce8a90b · ce8a90b
1 parent ecb434d
commit ce8a90b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/compiler/src/dotty/tools/dotc/core/TypeApplications.scala b/compiler/src/dotty/tools/dotc/core/TypeApplications.scala
@@ -506,10 +506,11 @@ class TypeApplications(val self: Type) extends AnyVal {
     case _ => Nil
 
   /** If this is an encoding of a function type, return its arguments, otherwise return Nil.
-   *  Handles ErasedFunctions gracefully.
+   *  Handles `ErasedFunction`s and poly functions gracefully.
    */
   final def functionArgInfos(using Context): List[Type] = self.dealias match
     case RefinedType(parent, nme.apply, mt: MethodType) if defn.isErasedFunctionType(parent) => (mt.paramInfos :+ mt.resultType)
+    case RefinedType(parent, nme.apply, mt: MethodType) if parent.typeSymbol eq defn.PolyFunctionClass => (mt.paramInfos :+ mt.resultType)
     case _ => self.dropDependentRefinement.dealias.argInfos
 
   /** Argument types where existential types in arguments are disallowed */