text: # coding=utf-8
  # --------------------------------------------------------------------------
  # Copyright (c) Microsoft Corporation. All rights reserved.
  # Licensed under the MIT License. See License.txt in the project root for license information.
  # Code generated by Microsoft (R) AutoRest Code Generator.
  # Changes may cause incorrect behavior and will be lost if the code is regenerated.
  # --------------------------------------------------------------------------
  
  from copy import deepcopy
  from typing import Any, Awaitable, Optional, TYPE_CHECKING
  
  from azure.core.rest import AsyncHttpResponse, HttpRequest
  from azure.mgmt.core import AsyncARMPipelineClient
  from msrest import Deserializer, Serializer
  
  from .. import models
  from ._configuration import AzureMachineLearningWorkspacesConfiguration
  from .operations import CodeContainersOperations, CodeVersionsOperations, ComponentContainersOperations, ComponentVersionsOperations, EnvironmentContainersOperations, EnvironmentVersionsOperations, ModelContainersOperations, ModelVersionsOperations, TemporaryDataReferencesOperations
  
  if TYPE_CHECKING:
      # pylint: disable=unused-import,ungrouped-imports
      from azure.core.credentials_async import AsyncTokenCredential
  
  class AzureMachineLearningWorkspaces:
      """AzureMachineLearningWorkspaces.
  
      :ivar code_containers: CodeContainersOperations operations
      :vartype code_containers:
       azure.mgmt.machinelearningservices.aio.operations.CodeContainersOperations
      :ivar code_versions: CodeVersionsOperations operations
      :vartype code_versions:
       azure.mgmt.machinelearningservices.aio.operations.CodeVersionsOperations
      :ivar component_containers: ComponentContainersOperations operations
      :vartype component_containers:
       azure.mgmt.machinelearningservices.aio.operations.ComponentContainersOperations
      :ivar component_versions: ComponentVersionsOperations operations
      :vartype component_versions:
       azure.mgmt.machinelearningservices.aio.operations.ComponentVersionsOperations
      :ivar environment_containers: EnvironmentContainersOperations operations
      :vartype environment_containers:
       azure.mgmt.machinelearningservices.aio.operations.EnvironmentContainersOperations
      :ivar environment_versions: EnvironmentVersionsOperations operations
      :vartype environment_versions:
       azure.mgmt.machinelearningservices.aio.operations.EnvironmentVersionsOperations
      :ivar model_containers: ModelContainersOperations operations
      :vartype model_containers:
       azure.mgmt.machinelearningservices.aio.operations.ModelContainersOperations
      :ivar model_versions: ModelVersionsOperations operations
      :vartype model_versions:
       azure.mgmt.machinelearningservices.aio.operations.ModelVersionsOperations
      :ivar temporary_data_references: TemporaryDataReferencesOperations operations
      :vartype temporary_data_references:
       azure.mgmt.machinelearningservices.aio.operations.TemporaryDataReferencesOperations
      :param credential: Credential needed for the client to connect to Azure.
      :type credential: ~azure.core.credentials_async.AsyncTokenCredential
      :param subscription_id: The ID of the target subscription.
      :type subscription_id: str
      :param base_url: Service URL. Default value is 'https://management.azure.com'.
      :type base_url: str
      :keyword api_version: Api Version. The default value is "2021-10-01-dataplanepreview". Note
       that overriding this default value may result in unsupported behavior.
      :paramtype api_version: str
      :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
       Retry-After header is present.
      """
  
      def __init__(
          self,
          credential: "AsyncTokenCredential",
          subscription_id: str,
          base_url: str = "https://management.azure.com",
          **kwargs: Any
      ) -> None:
          self._config = AzureMachineLearningWorkspacesConfiguration(credential=credential, subscription_id=subscription_id, **kwargs)
          self._client = AsyncARMPipelineClient(base_url=base_url, config=self._config, **kwargs)
  
          client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
          self._serialize = Serializer(client_models)
          self._deserialize = Deserializer(client_models)
          self._serialize.client_side_validation = False
          self.code_containers = CodeContainersOperations(self._client, self._config, self._serialize, self._deserialize)
          self.code_versions = CodeVersionsOperations(self._client, self._config, self._serialize, self._deserialize)
          self.component_containers = ComponentContainersOperations(self._client, self._config, self._serialize, self._deserialize)
          self.component_versions = ComponentVersionsOperations(self._client, self._config, self._serialize, self._deserialize)
          self.environment_containers = EnvironmentContainersOperations(self._client, self._config, self._serialize, self._deserialize)
          self.environment_versions = EnvironmentVersionsOperations(self._client, self._config, self._serialize, self._deserialize)
          self.model_containers = ModelContainersOperations(self._client, self._config, self._serialize, self._deserialize)
          self.model_versions = ModelVersionsOperations(self._client, self._config, self._serialize, self._deserialize)
          self.temporary_data_references = TemporaryDataReferencesOperations(self._client, self._config, self._serialize, self._deserialize)
  
  
      def _send_request(
          self,
          request: HttpRequest,
          **kwargs: Any
      ) -> Awaitable[AsyncHttpResponse]:
          """Runs the network request through the client's chained policies.
  
          >>> from azure.core.rest import HttpRequest
          >>> request = HttpRequest("GET", "https://www.example.org/")
          <HttpRequest [GET], url: 'https://www.example.org/'>
          >>> response = await client._send_request(request)
          <AsyncHttpResponse: 200 OK>
  
          For more information on this code flow, see https://aka.ms/azsdk/python/protocol/quickstart
  
          :param request: The network request you want to make. Required.
          :type request: ~azure.core.rest.HttpRequest
          :keyword bool stream: Whether the response payload will be streamed. Defaults to False.
          :return: The response of your network call. Does not do error handling on your response.
          :rtype: ~azure.core.rest.AsyncHttpResponse
          """
  
          request_copy = deepcopy(request)
          request_copy.url = self._client.format_url(request_copy.url)
          return self._client.send_request(request_copy, **kwargs)
  
      async def close(self) -> None:
          await self._client.close()
  
      async def __aenter__(self) -> "AzureMachineLearningWorkspaces":
          await self._client.__aenter__()
          return self
  
      async def __aexit__(self, *exc_details) -> None:
          await self._client.__aexit__(*exc_details)
  