Python raise_mg_import_exception示例

示例#1

文件： kneighbors_classifier.py 项目： risingsudhir/cuml

    def _func_create_model(sessionId, **kwargs):
        try:
            from cuml.neighbors.kneighbors_classifier_mg import \
                KNeighborsClassifierMG as cumlKNN
        except ImportError:
            raise_mg_import_exception()

        handle = worker_state(sessionId)["handle"]
        return cumlKNN(handle=handle, **kwargs)

示例#2

    def _func_create_model(sessionId, **kwargs):
        try:
            from cuml.neighbors.kneighbors_regressor_mg import \
                KNeighborsRegressorMG as cumlKNN
        except ImportError:
            raise_mg_import_exception()

        handle = get_raft_comm_state(sessionId)["handle"]
        return cumlKNN(handle=handle, **kwargs)