From a92e844dd1b85bce48cd7838e82ebaa16dad983f Mon Sep 17 00:00:00 2001
From: Krista Pratico <krpratic@microsoft.com>
Date: Mon, 13 Jul 2020 08:42:27 -0700
Subject: [PATCH] fix mypy for release/v3 branch

---
 .../azure/ai/formrecognizer/_form_training_client.py        | 6 +++---
 .../ai/formrecognizer/aio/_form_training_client_async.py    | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_training_client.py b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_training_client.py
index b79480d379d2..8dd8dc11f9e9 100644
--- a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_training_client.py
+++ b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/_form_training_client.py
@@ -9,7 +9,6 @@
 import json
 from typing import (
     Any,
-    Iterable,
     Dict,
     Union,
     TYPE_CHECKING,
@@ -39,6 +38,7 @@
     from azure.core.credentials import AzureKeyCredential, TokenCredential
     from azure.core.pipeline import PipelineResponse
     from azure.core.pipeline.transport import HttpResponse
+    from azure.core.paging import ItemPaged
     PipelineResponseType = HttpResponse
 
 
@@ -182,7 +182,7 @@ def delete_model(self, model_id, **kwargs):
 
     @distributed_trace
     def list_custom_models(self, **kwargs):
-        # type: (Any) -> Iterable[CustomFormModelInfo]
+        # type: (Any) -> ItemPaged[CustomFormModelInfo]
         """List information for each model, including model id,
         model status, and when it was created and last modified.
 
@@ -199,7 +199,7 @@ def list_custom_models(self, **kwargs):
                 :dedent: 8
                 :caption: List model information for each model on the account.
         """
-        return self._client.list_custom_models(
+        return self._client.list_custom_models(  # type: ignore
             cls=kwargs.pop("cls", lambda objs: [CustomFormModelInfo._from_generated(x) for x in objs]),
             error_map=error_map,
             **kwargs
diff --git a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_training_client_async.py b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_training_client_async.py
index 70611586bf83..c985d303a94e 100644
--- a/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_training_client_async.py
+++ b/sdk/formrecognizer/azure-ai-formrecognizer/azure/ai/formrecognizer/aio/_form_training_client_async.py
@@ -9,7 +9,6 @@
 import json
 from typing import (
     Any,
-    AsyncIterable,
     Dict,
     Union,
     TYPE_CHECKING,
@@ -18,6 +17,7 @@
 from azure.core.polling.async_base_polling import AsyncLROBasePolling
 from azure.core.tracing.decorator import distributed_trace
 from azure.core.tracing.decorator_async import distributed_trace_async
+from azure.core.async_paging import AsyncItemPaged
 from ._form_recognizer_client_async import FormRecognizerClient
 from .._generated.aio._form_recognizer_client_async import FormRecognizerClient as FormRecognizer
 from .._generated.models import (
@@ -188,7 +188,7 @@ async def delete_model(self, model_id: str, **kwargs: Any) -> None:
         )
 
     @distributed_trace
-    def list_custom_models(self, **kwargs: Any) -> AsyncIterable[CustomFormModelInfo]:
+    def list_custom_models(self, **kwargs: Any) -> AsyncItemPaged[CustomFormModelInfo]:
         """List information for each model, including model id,
         model status, and when it was created and last modified.
 
@@ -205,7 +205,7 @@ def list_custom_models(self, **kwargs: Any) -> AsyncIterable[CustomFormModelInfo
                 :dedent: 12
                 :caption: List model information for each model on the account.
         """
-        return self._client.list_custom_models(
+        return self._client.list_custom_models(  # type: ignore
             cls=kwargs.pop("cls", lambda objs: [CustomFormModelInfo._from_generated(x) for x in objs]),
             error_map=error_map,
             **kwargs