fix lint and mypy errors

sdk/communication/azure-communication-jobrouter/azure/communication/jobrouter/_operations/_operations.py

@@ -2146,8 +2146,9 @@ def get_distribution_policy(self, distribution_policy_id: str, **kwargs: Any) ->
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
-    def list_distribution_policies(self, **kwargs: Any) -> Iterable["_models.DistributionPolicy"]:
+    def list_distribution_policies(self, **kwargs: Any) -> ItemPaged["_models.DistributionPolicy"]:
         # pylint: disable=line-too-long
         """Retrieves existing distribution policies.
 
@@ -3340,8 +3341,9 @@ def get_classification_policy(self, classification_policy_id: str, **kwargs: Any
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
-    def list_classification_policies(self, **kwargs: Any) -> Iterable["_models.ClassificationPolicy"]:
+    def list_classification_policies(self, **kwargs: Any) -> ItemPaged["_models.ClassificationPolicy"]:
         # pylint: disable=line-too-long
         """Retrieves existing classification policies.
 
@@ -3963,8 +3965,9 @@ def get_exception_policy(self, exception_policy_id: str, **kwargs: Any) -> _mode
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
-    def list_exception_policies(self, **kwargs: Any) -> Iterable["_models.ExceptionPolicy"]:
+    def list_exception_policies(self, **kwargs: Any) -> ItemPaged["_models.ExceptionPolicy"]:
         """Retrieves existing exception policies.
 
         Retrieves existing exception policies.
@@ -4539,8 +4542,9 @@ def get_queue(self, queue_id: str, **kwargs: Any) -> _models.RouterQueue:
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
-    def list_queues(self, **kwargs: Any) -> Iterable["_models.RouterQueue"]:
+    def list_queues(self, **kwargs: Any) -> ItemPaged["_models.RouterQueue"]:
         # pylint: disable=line-too-long
         """Retrieves existing queues.
 
@@ -6423,6 +6427,7 @@ def _close_job(  # pylint: disable=protected-access
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
     def list_jobs(
         self,
@@ -6434,7 +6439,7 @@ def list_jobs(
         scheduled_before: Optional[datetime.datetime] = None,
         scheduled_after: Optional[datetime.datetime] = None,
         **kwargs: Any
-    ) -> Iterable["_models.RouterJob"]:
+    ) -> ItemPaged["_models.RouterJob"]:
         # pylint: disable=line-too-long
         """Retrieves list of jobs based on filter parameters.
 
@@ -8039,6 +8044,7 @@ def delete_worker(self, worker_id: str, **kwargs: Any) -> None:  # pylint: disab
         if cls:
             return cls(pipeline_response, None, {})  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
     def list_workers(
         self,
@@ -8048,7 +8054,7 @@ def list_workers(
         queue_id: Optional[str] = None,
         has_capacity: Optional[bool] = None,
         **kwargs: Any
-    ) -> Iterable["_models.RouterWorker"]:
+    ) -> ItemPaged["_models.RouterWorker"]:
         # pylint: disable=line-too-long
         """Retrieves existing workers.
 

sdk/communication/azure-communication-jobrouter/azure/communication/jobrouter/_operations/_patch.py

@@ -169,7 +169,7 @@ def upsert_exception_policy(
     def upsert_exception_policy(
         self,
         exception_policy_id: str,
-        resource: IO,
+        resource: IO[bytes],
         *,
         content_type: str = "application/merge-patch+json",
         if_unmodified_since: Optional[datetime.datetime] = None,
@@ -184,7 +184,7 @@ def upsert_exception_policy(
         :param exception_policy_id: The Id of the exception policy. Required.
         :type exception_policy_id: str
         :param resource: The resource instance. Required.
-        :type resource: IO
+        :type resource: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/merge-patch+json".
         :paramtype content_type: str
@@ -204,7 +204,7 @@ def upsert_exception_policy(
         """
 
     def upsert_exception_policy(
-        self, exception_policy_id: str, *args: Union[_models.ExceptionPolicy, JSON, IO], **kwargs: Any
+        self, exception_policy_id: str, *args: Union[_models.ExceptionPolicy, JSON, IO[bytes]], **kwargs: Any
     ) -> _models.ExceptionPolicy:
         """Update an exception policy.
 
@@ -242,7 +242,7 @@ def upsert_exception_policy(
         if not exception_policy_id:
             raise ValueError("exception_policy_id cannot be None.")
 
-        exception_policy: Union[_models.ExceptionPolicy, JSON, IO] = _models.ExceptionPolicy()
+        exception_policy: Union[_models.ExceptionPolicy, JSON, IO[bytes]] = _models.ExceptionPolicy()
         if len(args) == 1:
             exception_policy = args[0]
 
@@ -431,7 +431,7 @@ def upsert_distribution_policy(
     def upsert_distribution_policy(
         self,
         distribution_policy_id: str,
-        resource: IO,
+        resource: IO[bytes],
         *,
         content_type: str = "application/merge-patch+json",
         if_unmodified_since: Optional[datetime.datetime] = None,
@@ -446,7 +446,7 @@ def upsert_distribution_policy(
         :param distribution_policy_id: The unique identifier of the policy. Required.
         :type distribution_policy_id: str
         :param resource: The resource instance. Required.
-        :type resource: IO
+        :type resource: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/merge-patch+json".
         :paramtype content_type: str
@@ -466,7 +466,7 @@ def upsert_distribution_policy(
         """
 
     def upsert_distribution_policy(
-        self, distribution_policy_id: str, *args: Union[_models.DistributionPolicy, JSON, IO], **kwargs: Any
+        self, distribution_policy_id: str, *args: Union[_models.DistributionPolicy, JSON, IO[bytes]], **kwargs: Any
     ) -> _models.DistributionPolicy:
         """Update a distribution policy.
 
@@ -511,7 +511,7 @@ def upsert_distribution_policy(
         if not distribution_policy_id:
             raise ValueError("distribution_policy_id cannot be None.")
 
-        distribution_policy: Union[_models.DistributionPolicy, JSON, IO] = _models.DistributionPolicy()
+        distribution_policy: Union[_models.DistributionPolicy, JSON, IO[bytes]] = _models.DistributionPolicy()
         if len(args) == 1:
             distribution_policy = args[0]
 
@@ -710,7 +710,7 @@ def upsert_queue(
     def upsert_queue(
         self,
         queue_id: str,
-        resource: IO,
+        resource: IO[bytes],
         *,
         content_type: str = "application/merge-patch+json",
         if_unmodified_since: Optional[datetime.datetime] = None,
@@ -725,7 +725,7 @@ def upsert_queue(
         :param queue_id: The Id of this queue. Required.
         :type queue_id: str
         :param resource: The resource instance. Required.
-        :type resource: IO
+        :type resource: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/merge-patch+json".
         :paramtype content_type: str
@@ -745,7 +745,7 @@ def upsert_queue(
         """
 
     def upsert_queue(
-        self, queue_id: str, *args: Union[_models.RouterQueue, JSON, IO], **kwargs: Any
+        self, queue_id: str, *args: Union[_models.RouterQueue, JSON, IO[bytes]], **kwargs: Any
     ) -> _models.RouterQueue:
         """Update a job queue
 
@@ -792,7 +792,7 @@ def upsert_queue(
         if not queue_id:
             raise ValueError("queue_id cannot be None.")
 
-        queue: Union[_models.RouterQueue, JSON, IO] = _models.RouterQueue()
+        queue: Union[_models.RouterQueue, JSON, IO[bytes]] = _models.RouterQueue()
         if len(args) == 1:
             queue = args[0]
 
@@ -1027,7 +1027,7 @@ def upsert_classification_policy(
     def upsert_classification_policy(
         self,
         classification_policy_id: str,
-        resource: IO,
+        resource: IO[bytes],
         *,
         content_type: str = "application/merge-patch+json",
         if_unmodified_since: Optional[datetime.datetime] = None,
@@ -1042,7 +1042,7 @@ def upsert_classification_policy(
         :param classification_policy_id: Unique identifier of this policy. Required.
         :type classification_policy_id: str
         :param resource: The resource instance. Required.
-        :type resource: IO
+        :type resource: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/merge-patch+json".
         :paramtype content_type: str
@@ -1062,7 +1062,7 @@ def upsert_classification_policy(
         """
 
     def upsert_classification_policy(
-        self, classification_policy_id: str, *args: Union[_models.ClassificationPolicy, JSON, IO], **kwargs: Any
+        self, classification_policy_id: str, *args: Union[_models.ClassificationPolicy, JSON, IO[bytes]], **kwargs: Any
     ) -> _models.ClassificationPolicy:
         """Update a classification policy
 
@@ -1120,7 +1120,7 @@ def upsert_classification_policy(
         if not classification_policy_id:
             raise ValueError("classification_policy_id cannot be None.")
 
-        classification_policy: Union[_models.ClassificationPolicy, JSON, IO] = _models.ClassificationPolicy()
+        classification_policy: Union[_models.ClassificationPolicy, JSON, IO[bytes]] = _models.ClassificationPolicy()
         if len(args) == 1:
             classification_policy = args[0]
 
@@ -1356,7 +1356,7 @@ def upsert_worker(
     def upsert_worker(
         self,
         worker_id: str,
-        resource: IO,
+        resource: IO[bytes],
         *,
         content_type: str = "application/merge-patch+json",
         if_unmodified_since: Optional[datetime.datetime] = None,
@@ -1371,7 +1371,7 @@ def upsert_worker(
         :param worker_id: Id of the worker. Required.
         :type worker_id: str
         :param resource: The resource instance. Required.
-        :type resource: IO
+        :type resource: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/merge-patch+json".
         :paramtype content_type: str
@@ -1391,7 +1391,7 @@ def upsert_worker(
         """
 
     def upsert_worker(
-        self, worker_id: str, *args: Union[_models.RouterWorker, JSON, IO], **kwargs: Any
+        self, worker_id: str, *args: Union[_models.RouterWorker, JSON, IO[bytes]], **kwargs: Any
     ) -> _models.RouterWorker:
         """Update a router worker.
 
@@ -1462,11 +1462,20 @@ def upsert_worker(
                 :language: python
                 :dedent: 8
                 :caption: Use a JobRouterClient to de-register a worker
+
+        .. admonition:: Example:
+
+            .. literalinclude:: ../samples/router_worker_crud_ops.py
+                :start-after: [START create_worker_w_limit_concurrent_offers]
+                :end-before: [END create_worker_w_limit_concurrent_offers]
+                :language: python
+                :dedent: 8
+                :caption: Use a JobRouterClient to create a worker with concurrent offer limit
         """
         if not worker_id:
             raise ValueError("worker_id cannot be None.")
 
-        router_worker: Union[_models.RouterWorker, JSON, IO] = _models.RouterWorker()
+        router_worker: Union[_models.RouterWorker, JSON, IO[bytes]] = _models.RouterWorker()
         if len(args) == 1:
             router_worker = args[0]
 
@@ -1725,7 +1734,7 @@ def upsert_job(
     def upsert_job(
         self,
         job_id: str,
-        resource: IO,
+        resource: IO[bytes],
         *,
         content_type: str = "application/merge-patch+json",
         if_unmodified_since: Optional[datetime.datetime] = None,
@@ -1740,7 +1749,7 @@ def upsert_job(
         :param job_id: The id of the job. Required.
         :type job_id: str
         :param resource: The resource instance. Required.
-        :type resource: IO
+        :type resource: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/merge-patch+json".
         :paramtype content_type: str
@@ -1759,7 +1768,12 @@ def upsert_job(
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
-    def upsert_job(self, job_id: str, *args: Union[_models.RouterJob, JSON, IO], **kwargs: Any) -> _models.RouterJob:
+    def upsert_job(
+            self,
+            job_id: str,
+            *args: Union[_models.RouterJob, JSON, IO[bytes]],
+            **kwargs: Any
+    ) -> _models.RouterJob:
         """Update a job.
 
         :param str job_id: Id of the job.
@@ -1826,7 +1840,7 @@ def upsert_job(self, job_id: str, *args: Union[_models.RouterJob, JSON, IO], **k
         if not job_id:
             raise ValueError("job_id cannot be None.")
 
-        router_job: Union[_models.RouterJob, JSON, IO] = _models.RouterJob()
+        router_job: Union[_models.RouterJob, JSON, IO[bytes]] = _models.RouterJob()
         if len(args) == 1:
             router_job = args[0]
 

sdk/communication/azure-communication-jobrouter/azure/communication/jobrouter/aio/_operations/_operations.py

@@ -1237,8 +1237,9 @@ async def get_distribution_policy(self, distribution_policy_id: str, **kwargs: A
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
-    def list_distribution_policies(self, **kwargs: Any) -> AsyncIterable["_models.DistributionPolicy"]:
+    def list_distribution_policies(self, **kwargs: Any) -> AsyncItemPaged["_models.DistributionPolicy"]:
         # pylint: disable=line-too-long
         """Retrieves existing distribution policies.
 
@@ -2434,8 +2435,9 @@ async def get_classification_policy(
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
-    def list_classification_policies(self, **kwargs: Any) -> AsyncIterable["_models.ClassificationPolicy"]:
+    def list_classification_policies(self, **kwargs: Any) -> AsyncItemPaged["_models.ClassificationPolicy"]:
         # pylint: disable=line-too-long
         """Retrieves existing classification policies.
 
@@ -3057,8 +3059,9 @@ async def get_exception_policy(self, exception_policy_id: str, **kwargs: Any) ->
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
-    def list_exception_policies(self, **kwargs: Any) -> AsyncIterable["_models.ExceptionPolicy"]:
+    def list_exception_policies(self, **kwargs: Any) -> AsyncItemPaged["_models.ExceptionPolicy"]:
         """Retrieves existing exception policies.
 
         Retrieves existing exception policies.
@@ -3634,8 +3637,9 @@ async def get_queue(self, queue_id: str, **kwargs: Any) -> _models.RouterQueue:
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
-    def list_queues(self, **kwargs: Any) -> AsyncIterable["_models.RouterQueue"]:
+    def list_queues(self, **kwargs: Any) -> AsyncItemPaged["_models.RouterQueue"]:
         # pylint: disable=line-too-long
         """Retrieves existing queues.
 
@@ -5521,6 +5525,7 @@ async def _close_job(  # pylint: disable=protected-access
 
         return deserialized  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
     def list_jobs(
         self,
@@ -5532,7 +5537,7 @@ def list_jobs(
         scheduled_before: Optional[datetime.datetime] = None,
         scheduled_after: Optional[datetime.datetime] = None,
         **kwargs: Any
-    ) -> AsyncIterable["_models.RouterJob"]:
+    ) -> AsyncItemPaged["_models.RouterJob"]:
         # pylint: disable=line-too-long
         """Retrieves list of jobs based on filter parameters.
 
@@ -7140,6 +7145,7 @@ async def delete_worker(  # pylint: disable=inconsistent-return-statements
         if cls:
             return cls(pipeline_response, None, {})  # type: ignore
 
+    # https://github.com/Azure/autorest.python/issues/2262
     @distributed_trace
     def list_workers(
         self,
@@ -7149,7 +7155,7 @@ def list_workers(
         queue_id: Optional[str] = None,
         has_capacity: Optional[bool] = None,
         **kwargs: Any
-    ) -> AsyncIterable["_models.RouterWorker"]:
+    ) -> AsyncItemPaged["_models.RouterWorker"]:
         # pylint: disable=line-too-long
         """Retrieves existing workers.