@@ -334,6 +334,14 @@ def _from_rest_object(cls, rest_obj: RestWorkspace) -> Optional["Workspace"]:
334
334
if hasattr (rest_obj , "ml_flow_tracking_uri" ):
335
335
mlflow_tracking_uri = rest_obj .ml_flow_tracking_uri
336
336
337
+ # TODO: Remove once Online endpoints updates API version to at least 2023-08-01
338
+ allow_roleassignment_on_rg = None
339
+ if hasattr (rest_obj , "allow_roleassignment_on_rg" ):
340
+ allow_roleassignment_on_rg = rest_obj .allow_roleassignment_on_rg
341
+ system_datastores_auth_mode = None
342
+ if hasattr (rest_obj , "system_datastores_auth_mode" ):
343
+ system_datastores_auth_mode = rest_obj .system_datastores_auth_mode
344
+
337
345
# TODO: remove this once it is included in API response
338
346
managed_network = None
339
347
if hasattr (rest_obj , "managed_network" ):
@@ -386,10 +394,10 @@ def _from_rest_object(cls, rest_obj: RestWorkspace) -> Optional["Workspace"]:
386
394
identity = identity ,
387
395
primary_user_assigned_identity = rest_obj .primary_user_assigned_identity ,
388
396
managed_network = managed_network ,
389
- system_datastores_auth_mode = rest_obj . system_datastores_auth_mode ,
397
+ system_datastores_auth_mode = system_datastores_auth_mode ,
390
398
feature_store_settings = feature_store_settings ,
391
399
enable_data_isolation = rest_obj .enable_data_isolation ,
392
- allow_roleassignment_on_rg = rest_obj . allow_roleassignment_on_rg ,
400
+ allow_roleassignment_on_rg = allow_roleassignment_on_rg ,
393
401
hub_id = rest_obj .hub_resource_id ,
394
402
workspace_id = rest_obj .workspace_id ,
395
403
serverless_compute = serverless_compute ,
0 commit comments