Fixed a bug that resulted in incorrect type evaluation when an async …

…function with an inferred return type is decorated with a class or function that uses a ParamSpec (such as `functools.wraps`). This addresses #6080. (#6081) Co-authored-by: Eric Traut <[email protected]>
microsoft · Oct 4, 2023 · b7bbbd6 · b7bbbd6
1 parent 4137f0c
commit b7bbbd6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/packages/pyright-internal/src/analyzer/typeUtils.ts b/packages/pyright-internal/src/analyzer/typeUtils.ts
@@ -2584,7 +2584,10 @@ export function removeParamSpecVariadicsFromFunction(type: FunctionType): Functi
         return type;
     }
 
-    return FunctionType.cloneRemoveParamSpecVariadics(type, argsParam.type);
+    return FunctionType.cloneRemoveParamSpecVariadics(
+        type,
+        TypeVarType.cloneForParamSpecAccess(argsParam.type, /* access */ undefined)
+    );
 }
 
 function _expandVariadicUnpackedUnion(type: Type) {

diff --git a/packages/pyright-internal/src/analyzer/types.ts b/packages/pyright-internal/src/analyzer/types.ts
@@ -1848,6 +1848,10 @@ export namespace FunctionType {
             newFunction.details.paramSpec = paramSpec;
         }
 
+        if (type.inferredReturnType) {
+            newFunction.inferredReturnType = type.inferredReturnType;
+        }
+
         return newFunction;
     }