fix uint as discrete column (#277)

mljar · Jan 11, 2021 · bcc280e · bcc280e
1 parent 1d868b0
commit bcc280e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/supervised/preprocessing/preprocessing_utils.py b/supervised/preprocessing/preprocessing_utils.py
@@ -27,7 +27,7 @@ def get_type(x):
         data_type = PreprocessingUtils.CATEGORICAL
         if col_type.startswith("float"):
             data_type = PreprocessingUtils.CONTINOUS
-        elif col_type.startswith("int"):
+        elif col_type.startswith("int") or col_type.startswith("uint"):
             data_type = PreprocessingUtils.DISCRETE
         elif col_type.startswith("datetime"):
             data_type = PreprocessingUtils.DATETIME