diff --git a/sdk/core/azure-core/azure/core/polling/_async_poller.py b/sdk/core/azure-core/azure/core/polling/_async_poller.py index cb51a3b2289a..935dc455987e 100644 --- a/sdk/core/azure-core/azure/core/polling/_async_poller.py +++ b/sdk/core/azure-core/azure/core/polling/_async_poller.py @@ -84,7 +84,7 @@ async def async_poller(client, initial_response, deserialization_callback, polli :param client: A pipeline service client. :type client: ~azure.core.PipelineClient :param initial_response: The initial call response - :type initial_response: ~azure.core.pipeline.transport.AsyncHttpResponse + :type initial_response: ~azure.core.pipeline.PipelineResponse :param deserialization_callback: A callback that takes a Response and return a deserialized object. If a subclass of Model is given, this passes "deserialize" as callback. :type deserialization_callback: callable or msrest.serialization.Model @@ -101,8 +101,7 @@ class AsyncLROPoller(Generic[PollingReturnType], Awaitable): :param client: A pipeline service client :type client: ~azure.core.PipelineClient :param initial_response: The initial call response - :type initial_response: - ~azure.core.pipeline.transport.HttpResponse or ~azure.core.pipeline.transport.AsyncHttpResponse + :type initial_response: ~azure.core.pipeline.PipelineResponse :param deserialization_callback: A callback that takes a Response and return a deserialized object. If a subclass of Model is given, this passes "deserialize" as callback. :type deserialization_callback: callable or msrest.serialization.Model diff --git a/sdk/core/azure-core/azure/core/polling/_poller.py b/sdk/core/azure-core/azure/core/polling/_poller.py index 4dff2ab220f6..a9f670504859 100644 --- a/sdk/core/azure-core/azure/core/polling/_poller.py +++ b/sdk/core/azure-core/azure/core/polling/_poller.py @@ -32,7 +32,6 @@ from urllib.parse import urlparse from typing import Any, Callable, Union, List, Optional, Tuple, TypeVar, Generic -from azure.core.pipeline.transport._base import HttpResponse from azure.core.tracing.decorator import distributed_trace from azure.core.tracing.common import with_current_context @@ -140,8 +139,7 @@ class LROPoller(Generic[PollingReturnType]): :param client: A pipeline service client :type client: ~azure.core.PipelineClient :param initial_response: The initial call response - :type initial_response: - ~azure.core.pipeline.transport.HttpResponse or ~azure.core.pipeline.transport.AsyncHttpResponse + :type initial_response: ~azure.core.pipeline.PipelineResponse :param deserialization_callback: A callback that takes a Response and return a deserialized object. If a subclass of Model is given, this passes "deserialize" as callback. :type deserialization_callback: callable or msrest.serialization.Model @@ -150,7 +148,7 @@ class LROPoller(Generic[PollingReturnType]): """ def __init__(self, client, initial_response, deserialization_callback, polling_method): - # type: (Any, HttpResponse, Callable, PollingMethod[PollingReturnType]) -> None + # type: (Any, Any, Callable, PollingMethod[PollingReturnType]) -> None self._callbacks = [] # type: List[Callable] self._polling_method = polling_method