From b92221eb977d52eb7383e3237549fb0a1c9f30a8 Mon Sep 17 00:00:00 2001
From: Emanuele Giaquinta <emanuele.giaquinta@aiven.io>
Date: Mon, 27 Nov 2023 09:41:09 +0200
Subject: [PATCH] psycopg2: fix range annotations (#11071)

---
 stubs/psycopg2/psycopg2/_range.pyi | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/stubs/psycopg2/psycopg2/_range.pyi b/stubs/psycopg2/psycopg2/_range.pyi
index 131196c95..0b3247455 100644
--- a/stubs/psycopg2/psycopg2/_range.pyi
+++ b/stubs/psycopg2/psycopg2/_range.pyi
@@ -33,7 +33,7 @@ class Range:
     def __ge__(self, other: Range) -> bool: ...
 
 def register_range(
-    pgrange: str, pyrange: str | Range, conn_or_curs: connection | cursor, globally: bool = False
+    pgrange: str, pyrange: str | type[Range], conn_or_curs: connection | cursor, globally: bool = False
 ) -> RangeCaster: ...
 
 class RangeAdapter:
@@ -51,7 +51,12 @@ class RangeCaster:
     typecaster: Any
     array_typecaster: Any
     def __init__(
-        self, pgrange: str | RangeAdapter, pyrange: str | Range, oid: int, subtype_oid: int, array_oid: int | None = None
+        self,
+        pgrange: str | type[RangeAdapter],
+        pyrange: str | type[Range],
+        oid: int,
+        subtype_oid: int,
+        array_oid: int | None = None,
     ) -> None: ...
     @overload
     def parse(self, s: None, cur: cursor | None = None) -> None: ...