ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-16 10:58:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

MI now supports ONNX runtime so the extension should support is as well (#10604)

Browse Source
This commit is contained in:
Leila Lali
2020-05-28 13:36:16 -07:00
committed by GitHub
parent 6cf8f9ff35
commit 0d544660e0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions/machine-learning/src/prediction/predictService.ts
View File

@@ -43,9 +43,9 @@ export class PredictService {
let connection = await this.getCurrentConnection();
if (connection) {
const serverInfo = await this._apiWrapper.getServerInfo(connection.connectionId);
// Right now only Azure SQL Edge support Onnx
// Right now only Azure SQL Edge and MI support Onnx
//
return serverInfo && serverInfo.engineEditionId === 9;
return serverInfo && (serverInfo.engineEditionId === 9 || serverInfo.engineEditionId === 8);
}
return false;
} catch (error) {