99from copy import deepcopy
1010from typing import Any , TYPE_CHECKING
1111
12+ from azure .core .pipeline import policies
1213from azure .core .rest import HttpRequest , HttpResponse
1314from azure .mgmt .core import ARMPipelineClient
15+ from azure .mgmt .core .policies import ARMAutoResourceProviderRegistrationPolicy
1416
1517from . import models as _models
1618from ._configuration import MachineLearningServicesMgmtClientConfiguration
2830 DatastoresOperations ,
2931 EnvironmentContainersOperations ,
3032 EnvironmentVersionsOperations ,
33+ FeaturesOperations ,
34+ FeaturesetContainersOperations ,
35+ FeaturesetVersionsOperations ,
36+ FeaturestoreEntityContainersOperations ,
37+ FeaturestoreEntityVersionsOperations ,
3138 JobsOperations ,
39+ ManagedNetworkProvisionsOperations ,
40+ ManagedNetworkSettingsRuleOperations ,
3241 ModelContainersOperations ,
3342 ModelVersionsOperations ,
3443 OnlineDeploymentsOperations ,
4352 RegistryComponentContainersOperations ,
4453 RegistryComponentVersionsOperations ,
4554 RegistryDataContainersOperations ,
55+ RegistryDataReferencesOperations ,
4656 RegistryDataVersionsOperations ,
4757 RegistryEnvironmentContainersOperations ,
4858 RegistryEnvironmentVersionsOperations ,
@@ -86,6 +96,12 @@ class MachineLearningServicesMgmtClient: # pylint: disable=client-accepts-api-v
8696 :ivar workspace_connections: WorkspaceConnectionsOperations operations
8797 :vartype workspace_connections:
8898 azure.mgmt.machinelearningservices.operations.WorkspaceConnectionsOperations
99+ :ivar managed_network_settings_rule: ManagedNetworkSettingsRuleOperations operations
100+ :vartype managed_network_settings_rule:
101+ azure.mgmt.machinelearningservices.operations.ManagedNetworkSettingsRuleOperations
102+ :ivar managed_network_provisions: ManagedNetworkProvisionsOperations operations
103+ :vartype managed_network_provisions:
104+ azure.mgmt.machinelearningservices.operations.ManagedNetworkProvisionsOperations
89105 :ivar registry_code_containers: RegistryCodeContainersOperations operations
90106 :vartype registry_code_containers:
91107 azure.mgmt.machinelearningservices.operations.RegistryCodeContainersOperations
@@ -104,6 +120,9 @@ class MachineLearningServicesMgmtClient: # pylint: disable=client-accepts-api-v
104120 :ivar registry_data_versions: RegistryDataVersionsOperations operations
105121 :vartype registry_data_versions:
106122 azure.mgmt.machinelearningservices.operations.RegistryDataVersionsOperations
123+ :ivar registry_data_references: RegistryDataReferencesOperations operations
124+ :vartype registry_data_references:
125+ azure.mgmt.machinelearningservices.operations.RegistryDataReferencesOperations
107126 :ivar registry_environment_containers: RegistryEnvironmentContainersOperations operations
108127 :vartype registry_environment_containers:
109128 azure.mgmt.machinelearningservices.operations.RegistryEnvironmentContainersOperations
@@ -146,6 +165,20 @@ class MachineLearningServicesMgmtClient: # pylint: disable=client-accepts-api-v
146165 :ivar environment_versions: EnvironmentVersionsOperations operations
147166 :vartype environment_versions:
148167 azure.mgmt.machinelearningservices.operations.EnvironmentVersionsOperations
168+ :ivar featureset_containers: FeaturesetContainersOperations operations
169+ :vartype featureset_containers:
170+ azure.mgmt.machinelearningservices.operations.FeaturesetContainersOperations
171+ :ivar features: FeaturesOperations operations
172+ :vartype features: azure.mgmt.machinelearningservices.operations.FeaturesOperations
173+ :ivar featureset_versions: FeaturesetVersionsOperations operations
174+ :vartype featureset_versions:
175+ azure.mgmt.machinelearningservices.operations.FeaturesetVersionsOperations
176+ :ivar featurestore_entity_containers: FeaturestoreEntityContainersOperations operations
177+ :vartype featurestore_entity_containers:
178+ azure.mgmt.machinelearningservices.operations.FeaturestoreEntityContainersOperations
179+ :ivar featurestore_entity_versions: FeaturestoreEntityVersionsOperations operations
180+ :vartype featurestore_entity_versions:
181+ azure.mgmt.machinelearningservices.operations.FeaturestoreEntityVersionsOperations
149182 :ivar jobs: JobsOperations operations
150183 :vartype jobs: azure.mgmt.machinelearningservices.operations.JobsOperations
151184 :ivar model_containers: ModelContainersOperations operations
@@ -172,7 +205,7 @@ class MachineLearningServicesMgmtClient: # pylint: disable=client-accepts-api-v
172205 :type subscription_id: str
173206 :param base_url: Service URL. Default value is "https://management.azure.com".
174207 :type base_url: str
175- :keyword api_version: Api Version. Default value is "2023 -04-01". Note that overriding this
208+ :keyword api_version: Api Version. Default value is "2024 -04-01". Note that overriding this
176209 default value may result in unsupported behavior.
177210 :paramtype api_version: str
178211 :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
@@ -189,7 +222,25 @@ def __init__(
189222 self ._config = MachineLearningServicesMgmtClientConfiguration (
190223 credential = credential , subscription_id = subscription_id , ** kwargs
191224 )
192- self ._client : ARMPipelineClient = ARMPipelineClient (base_url = base_url , config = self ._config , ** kwargs )
225+ _policies = kwargs .pop ("policies" , None )
226+ if _policies is None :
227+ _policies = [
228+ policies .RequestIdPolicy (** kwargs ),
229+ self ._config .headers_policy ,
230+ self ._config .user_agent_policy ,
231+ self ._config .proxy_policy ,
232+ policies .ContentDecodePolicy (** kwargs ),
233+ ARMAutoResourceProviderRegistrationPolicy (),
234+ self ._config .redirect_policy ,
235+ self ._config .retry_policy ,
236+ self ._config .authentication_policy ,
237+ self ._config .custom_hook_policy ,
238+ self ._config .logging_policy ,
239+ policies .DistributedTracingPolicy (** kwargs ),
240+ policies .SensitiveHeaderCleanupPolicy (** kwargs ) if self ._config .redirect_policy else None ,
241+ self ._config .http_logging_policy ,
242+ ]
243+ self ._client : ARMPipelineClient = ARMPipelineClient (base_url = base_url , policies = _policies , ** kwargs )
193244
194245 client_models = {k : v for k , v in _models .__dict__ .items () if isinstance (v , type )}
195246 self ._serialize = Serializer (client_models )
@@ -212,6 +263,12 @@ def __init__(
212263 self .workspace_connections = WorkspaceConnectionsOperations (
213264 self ._client , self ._config , self ._serialize , self ._deserialize
214265 )
266+ self .managed_network_settings_rule = ManagedNetworkSettingsRuleOperations (
267+ self ._client , self ._config , self ._serialize , self ._deserialize
268+ )
269+ self .managed_network_provisions = ManagedNetworkProvisionsOperations (
270+ self ._client , self ._config , self ._serialize , self ._deserialize
271+ )
215272 self .registry_code_containers = RegistryCodeContainersOperations (
216273 self ._client , self ._config , self ._serialize , self ._deserialize
217274 )
@@ -230,6 +287,9 @@ def __init__(
230287 self .registry_data_versions = RegistryDataVersionsOperations (
231288 self ._client , self ._config , self ._serialize , self ._deserialize
232289 )
290+ self .registry_data_references = RegistryDataReferencesOperations (
291+ self ._client , self ._config , self ._serialize , self ._deserialize
292+ )
233293 self .registry_environment_containers = RegistryEnvironmentContainersOperations (
234294 self ._client , self ._config , self ._serialize , self ._deserialize
235295 )
@@ -263,6 +323,19 @@ def __init__(
263323 self .environment_versions = EnvironmentVersionsOperations (
264324 self ._client , self ._config , self ._serialize , self ._deserialize
265325 )
326+ self .featureset_containers = FeaturesetContainersOperations (
327+ self ._client , self ._config , self ._serialize , self ._deserialize
328+ )
329+ self .features = FeaturesOperations (self ._client , self ._config , self ._serialize , self ._deserialize )
330+ self .featureset_versions = FeaturesetVersionsOperations (
331+ self ._client , self ._config , self ._serialize , self ._deserialize
332+ )
333+ self .featurestore_entity_containers = FeaturestoreEntityContainersOperations (
334+ self ._client , self ._config , self ._serialize , self ._deserialize
335+ )
336+ self .featurestore_entity_versions = FeaturestoreEntityVersionsOperations (
337+ self ._client , self ._config , self ._serialize , self ._deserialize
338+ )
266339 self .jobs = JobsOperations (self ._client , self ._config , self ._serialize , self ._deserialize )
267340 self .model_containers = ModelContainersOperations (
268341 self ._client , self ._config , self ._serialize , self ._deserialize
@@ -280,7 +353,7 @@ def __init__(
280353 self ._client , self ._config , self ._serialize , self ._deserialize
281354 )
282355
283- def _send_request (self , request : HttpRequest , ** kwargs : Any ) -> HttpResponse :
356+ def _send_request (self , request : HttpRequest , * , stream : bool = False , * *kwargs : Any ) -> HttpResponse :
284357 """Runs the network request through the client's chained policies.
285358
286359 >>> from azure.core.rest import HttpRequest
@@ -300,7 +373,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse:
300373
301374 request_copy = deepcopy (request )
302375 request_copy .url = self ._client .format_url (request_copy .url )
303- return self ._client .send_request (request_copy , ** kwargs )
376+ return self ._client .send_request (request_copy , stream = stream , ** kwargs ) # type: ignore
304377
305378 def close (self ) -> None :
306379 self ._client .close ()
0 commit comments