From 0d544660e0f989314c39311053dc29ba77bb047f Mon Sep 17 00:00:00 2001
From: Leila Lali <llali@microsoft.com>
Date: Thu, 28 May 2020 13:36:16 -0700
Subject: [PATCH] MI now supports ONNX runtime so the extension should support
 is as well (#10604)

---
 extensions/machine-learning/src/prediction/predictService.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/extensions/machine-learning/src/prediction/predictService.ts b/extensions/machine-learning/src/prediction/predictService.ts
index 600d5ff972..b3bec3a04a 100644
--- a/extensions/machine-learning/src/prediction/predictService.ts
+++ b/extensions/machine-learning/src/prediction/predictService.ts
@@ -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) {