Bring back Schema Registry to type complete (#36945)

* Bring back Schema Registry to type complete

* Explicit typing extensions dep

Author: lmazuel

sdk/schemaregistry/azure-schemaregistry/azure/schemaregistry/_patch.py

@@ -390,7 +390,7 @@ def __init__(self, **kwargs: Any) -> None:
         self.name: str = kwargs.pop("name")
         self.version: int = kwargs.pop("version")
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return (
             f"SchemaProperties(id={self.id}, format={self.format}, "
             f"group_name={self.group_name}, name={self.name}, version={self.version})"[:1024]
@@ -411,7 +411,7 @@ def __init__(self, **kwargs: Any) -> None:
         self.definition: str = kwargs.pop("definition")
         self.properties: SchemaProperties = kwargs.pop("properties")
 
-    def __repr__(self):
+    def __repr__(self) -> str:
         return f"Schema(definition={self.definition}, properties={self.properties})"[:1024]
 
 

sdk/schemaregistry/azure-schemaregistry/azure/schemaregistry/aio/_patch.py

@@ -18,6 +18,7 @@
     Dict,
 )
 from functools import partial
+from typing_extensions import Self
 
 from azure.core.tracing.decorator_async import distributed_trace_async
 
@@ -42,7 +43,7 @@
 ###### Wrapper Class ######
 
 
-class SchemaRegistryClient(object):
+class SchemaRegistryClient:
     """
     SchemaRegistryClient is a client for registering and retrieving schemas from the Azure Schema Registry service.
 
@@ -78,11 +79,11 @@ def __init__(
             **kwargs,
         )
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> Self:
         await self._generated_client.__aenter__()
         return self
 
-    async def __aexit__(self, *args):
+    async def __aexit__(self, *args: Any) -> None:
         await self._generated_client.__aexit__(*args)
 
     async def close(self) -> None:

sdk/schemaregistry/azure-schemaregistry/pyproject.toml

@@ -2,4 +2,3 @@
 mypy = true
 pyright = false
 type_check_samples = true
-verifytypes = false

sdk/schemaregistry/azure-schemaregistry/setup.py

@@ -70,6 +70,7 @@
     install_requires=[
         "azure-core<2.0.0,>=1.28.0",
         "isodate>=0.6.0",
+        "typing-extensions>=4.6.0"
     ],
     extras_require={
         "jsonencoder": [