From 296920a9e99ca4429e2f92d26ae407592af5ea27 Mon Sep 17 00:00:00 2001
From: Shantanu <12621235+hauntsaninja@users.noreply.github.com>
Date: Sat, 19 Feb 2022 19:57:43 -0800
Subject: [PATCH] zlib: update for py311 (#7315)

Co-authored-by: hauntsaninja <>
---
 stdlib/zlib.pyi | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/stdlib/zlib.pyi b/stdlib/zlib.pyi
index 2954a8402..cfd6784bb 100644
--- a/stdlib/zlib.pyi
+++ b/stdlib/zlib.pyi
@@ -1,3 +1,4 @@
+import sys
 from array import array
 from typing import Any
 from typing_extensions import Literal
@@ -41,7 +42,13 @@ class _Decompress:
     def copy(self) -> _Decompress: ...
 
 def adler32(__data: bytes, __value: int = ...) -> int: ...
-def compress(__data: bytes, level: int = ...) -> bytes: ...
+
+if sys.version_info >= (3, 11):
+    def compress(__data: bytes, level: int = ..., wbits: int = ...) -> bytes: ...
+
+else:
+    def compress(__data: bytes, level: int = ...) -> bytes: ...
+
 def compressobj(
     level: int = ..., method: int = ..., wbits: int = ..., memLevel: int = ..., strategy: int = ..., zdict: bytes | None = ...
 ) -> _Compress: ...