9
9
from copy import deepcopy
10
10
from typing import Any , TYPE_CHECKING
11
11
12
+ from azure .core .pipeline import policies
12
13
from azure .core .rest import HttpRequest , HttpResponse
13
14
from azure .mgmt .core import ARMPipelineClient
15
+ from azure .mgmt .core .policies import ARMAutoResourceProviderRegistrationPolicy
14
16
15
17
from . import models as _models
16
18
from ._configuration import MachineLearningServicesMgmtClientConfiguration
28
30
DatastoresOperations ,
29
31
EnvironmentContainersOperations ,
30
32
EnvironmentVersionsOperations ,
33
+ FeaturesOperations ,
34
+ FeaturesetContainersOperations ,
35
+ FeaturesetVersionsOperations ,
36
+ FeaturestoreEntityContainersOperations ,
37
+ FeaturestoreEntityVersionsOperations ,
31
38
JobsOperations ,
39
+ ManagedNetworkProvisionsOperations ,
40
+ ManagedNetworkSettingsRuleOperations ,
32
41
ModelContainersOperations ,
33
42
ModelVersionsOperations ,
34
43
OnlineDeploymentsOperations ,
43
52
RegistryComponentContainersOperations ,
44
53
RegistryComponentVersionsOperations ,
45
54
RegistryDataContainersOperations ,
55
+ RegistryDataReferencesOperations ,
46
56
RegistryDataVersionsOperations ,
47
57
RegistryEnvironmentContainersOperations ,
48
58
RegistryEnvironmentVersionsOperations ,
@@ -86,6 +96,12 @@ class MachineLearningServicesMgmtClient: # pylint: disable=client-accepts-api-v
86
96
:ivar workspace_connections: WorkspaceConnectionsOperations operations
87
97
:vartype workspace_connections:
88
98
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
89
105
:ivar registry_code_containers: RegistryCodeContainersOperations operations
90
106
:vartype registry_code_containers:
91
107
azure.mgmt.machinelearningservices.operations.RegistryCodeContainersOperations
@@ -104,6 +120,9 @@ class MachineLearningServicesMgmtClient: # pylint: disable=client-accepts-api-v
104
120
:ivar registry_data_versions: RegistryDataVersionsOperations operations
105
121
:vartype registry_data_versions:
106
122
azure.mgmt.machinelearningservices.operations.RegistryDataVersionsOperations
123
+ :ivar registry_data_references: RegistryDataReferencesOperations operations
124
+ :vartype registry_data_references:
125
+ azure.mgmt.machinelearningservices.operations.RegistryDataReferencesOperations
107
126
:ivar registry_environment_containers: RegistryEnvironmentContainersOperations operations
108
127
:vartype registry_environment_containers:
109
128
azure.mgmt.machinelearningservices.operations.RegistryEnvironmentContainersOperations
@@ -146,6 +165,20 @@ class MachineLearningServicesMgmtClient: # pylint: disable=client-accepts-api-v
146
165
:ivar environment_versions: EnvironmentVersionsOperations operations
147
166
:vartype environment_versions:
148
167
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
149
182
:ivar jobs: JobsOperations operations
150
183
:vartype jobs: azure.mgmt.machinelearningservices.operations.JobsOperations
151
184
:ivar model_containers: ModelContainersOperations operations
@@ -172,7 +205,7 @@ class MachineLearningServicesMgmtClient: # pylint: disable=client-accepts-api-v
172
205
:type subscription_id: str
173
206
:param base_url: Service URL. Default value is "https://management.azure.com".
174
207
: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
176
209
default value may result in unsupported behavior.
177
210
:paramtype api_version: str
178
211
:keyword int polling_interval: Default waiting time between two polls for LRO operations if no
@@ -189,7 +222,25 @@ def __init__(
189
222
self ._config = MachineLearningServicesMgmtClientConfiguration (
190
223
credential = credential , subscription_id = subscription_id , ** kwargs
191
224
)
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 )
193
244
194
245
client_models = {k : v for k , v in _models .__dict__ .items () if isinstance (v , type )}
195
246
self ._serialize = Serializer (client_models )
@@ -212,6 +263,12 @@ def __init__(
212
263
self .workspace_connections = WorkspaceConnectionsOperations (
213
264
self ._client , self ._config , self ._serialize , self ._deserialize
214
265
)
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
+ )
215
272
self .registry_code_containers = RegistryCodeContainersOperations (
216
273
self ._client , self ._config , self ._serialize , self ._deserialize
217
274
)
@@ -230,6 +287,9 @@ def __init__(
230
287
self .registry_data_versions = RegistryDataVersionsOperations (
231
288
self ._client , self ._config , self ._serialize , self ._deserialize
232
289
)
290
+ self .registry_data_references = RegistryDataReferencesOperations (
291
+ self ._client , self ._config , self ._serialize , self ._deserialize
292
+ )
233
293
self .registry_environment_containers = RegistryEnvironmentContainersOperations (
234
294
self ._client , self ._config , self ._serialize , self ._deserialize
235
295
)
@@ -263,6 +323,19 @@ def __init__(
263
323
self .environment_versions = EnvironmentVersionsOperations (
264
324
self ._client , self ._config , self ._serialize , self ._deserialize
265
325
)
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
+ )
266
339
self .jobs = JobsOperations (self ._client , self ._config , self ._serialize , self ._deserialize )
267
340
self .model_containers = ModelContainersOperations (
268
341
self ._client , self ._config , self ._serialize , self ._deserialize
@@ -280,7 +353,7 @@ def __init__(
280
353
self ._client , self ._config , self ._serialize , self ._deserialize
281
354
)
282
355
283
- def _send_request (self , request : HttpRequest , ** kwargs : Any ) -> HttpResponse :
356
+ def _send_request (self , request : HttpRequest , * , stream : bool = False , * *kwargs : Any ) -> HttpResponse :
284
357
"""Runs the network request through the client's chained policies.
285
358
286
359
>>> from azure.core.rest import HttpRequest
@@ -300,7 +373,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse:
300
373
301
374
request_copy = deepcopy (request )
302
375
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
304
377
305
378
def close (self ) -> None :
306
379
self ._client .close ()
0 commit comments