Fix not picking up return type with type params

Author: FlandiaYingman

hkmc2/shared/src/main/scala/hkmc2/syntax/Tree.scala

@@ -232,7 +232,7 @@ trait TypeOrTermDef:
       // fun f[T](n1: Int): Int
       // fun f[T](n1: Int)(nn: Int): Int
       case InfixApp(Apps(App(id: Ident, typeParams: TyTup), paramLists), Keyword.`:`, ret) =>
-        (symbName, R(id), paramLists, S(typeParams), N)
+        (symbName, R(id), paramLists, S(typeParams), S(ret))
 
       case InfixApp(Jux(lhs, rhs), Keyword.`:`, ret) =>
         rec(InfixApp(rhs, Keyword.`:`, ret), S(lhs))

hkmc2/shared/src/test/mlscript/basics/ModuleMethods.mls

@@ -34,7 +34,7 @@ module N with {
 module N with {
   fun f4[T](): module T = M
 }
-//│ ╔══[ERROR] Function returning module values must have explicit return types.
+//│ ╔══[ERROR] Function returning module values must have concrete return types.
 //│ ║  l.35: 	  fun f4[T](): module T = M
 //│ ╙──      	      ^^^^^^^^^^^^^^^^^