def _get_model_type(self):
    from .knn_utils import FAISSNeighborsClassifier, FAISSNeighborsRegressor
    if (self.problem_type == REGRESSION):
        return FAISSNeighborsRegressor
    else:
        return FAISSNeighborsClassifier
def _get_model_type(self):
    from .knn_utils import FAISSNeighborsClassifier, FAISSNeighborsRegressor
    if (self.problem_type == REGRESSION):
        return FAISSNeighborsRegressor
    else:
        return FAISSNeighborsClassifier