diff --git a/supervised/preprocessing/preprocessing_utils.py b/supervised/preprocessing/preprocessing_utils.py index a3202610..d4b43b83 100644 --- 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