ML - Removing reference to deprecated API (#11733)

* Removing reference to deprecated API
This commit is contained in:
Leila Lali
2020-08-10 13:06:47 -07:00
committed by GitHub
parent 858c1e826d
commit f7cd25af71
2 changed files with 6 additions and 9 deletions

View File

@@ -303,15 +303,12 @@ export class AzureModelRegistryService {
if (this._amlClient) {
return this._amlClient;
} else {
const tokens = await this._apiWrapper.getSecurityToken(account, azdata.AzureResource.ResourceManagement);
const tokens: { token: string, tokenType?: string } | undefined = await this._apiWrapper.getAccountSecurityToken(account, tenant.id, azdata.AzureResource.ResourceManagement);
let token: string = '';
let tokenType: string | undefined = undefined;
if (tokens && tenant.id in tokens) {
const tokenForId = tokens[tenant.id];
if (tokenForId) {
token = tokenForId.token;
tokenType = tokenForId.tokenType;
}
if (tokens) {
token = tokens.token;
tokenType = tokens.tokenType;
}
const client = new AzureMachineLearningWorkspaces(new TokenCredentials(token, tokenType), subscription.id, options);
if (apiVersion) {