ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-17 02:51:36 -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(); let connection = await this.getCurrentConnection();
if (connection) { if (connection) {
const serverInfo = await this._apiWrapper.getServerInfo(connection.connectionId); 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; return false;
} catch (error) { } catch (error) {