From 56d2a66ac5b0f8aba0eb1eb6ec73950e17ffe2ae Mon Sep 17 00:00:00 2001
From: Leila Lali <llali@microsoft.com>
Date: Wed, 7 Oct 2020 17:08:49 -0700
Subject: [PATCH] Fixed a regression in predict map columns page (#12799)

---
 .../src/views/models/prediction/columnsTable.ts                 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extensions/machine-learning/src/views/models/prediction/columnsTable.ts b/extensions/machine-learning/src/views/models/prediction/columnsTable.ts
index d843147431..218fe6ab9e 100644
--- a/extensions/machine-learning/src/views/models/prediction/columnsTable.ts
+++ b/extensions/machine-learning/src/views/models/prediction/columnsTable.ts
@@ -326,7 +326,7 @@ export class ColumnsTable extends ModelViewBase implements IDataComponent<Predic
 				}
 
 				const currentColumn = columns.find(x => x.columnName === value);
-				if (currentColumn && modelParameter.type !== currentColumn?.dataType) {
+				if (currentColumn && modelParameter.type === currentColumn?.dataType) {
 					inputContainer.removeItem(warningButton);
 				} else {
 					inputContainer.addItem(warningButton, {