import ray

from simdeep.simdeep_analysis import SimDeep

from simdeep.config import NB_CLUSTERS

from simdeep.config import CLUSTER_ARRAY

from simdeep.config import PVALUE_THRESHOLD

from simdeep.config import CINDEX_THRESHOLD

from simdeep.config import CLASSIFIER_TYPE

from simdeep.config import MIXTURE_PARAMS

from simdeep.config import PATH_RESULTS

from simdeep.config import PROJECT_NAME

from simdeep.config import CLASSIFICATION_METHOD

from simdeep.config import CLUSTER_EVAL_METHOD

from simdeep.config import CLUSTER_METHOD

from simdeep.config import NB_THREADS_COXPH

from simdeep.config import NB_SELECTED_FEATURES

from simdeep.config import LOAD_EXISTING_MODELS

from simdeep.config import NODES_SELECTION

from simdeep.config import PATH_TO_SAVE_MODEL

from simdeep.config import USE_AUTOENCODERS

from simdeep.config import FEATURE_SURV_ANALYSIS

from simdeep.config import CLUSTERING_OMICS

from simdeep.config import USE_R_PACKAGES_FOR_SURVIVAL

@ray.remote

class SimDeepDistributed(SimDeep):

    def __init__(

            self,

            nb_clusters=NB_CLUSTERS,

            pvalue_thres=PVALUE_THRESHOLD,

            cindex_thres=CINDEX_THRESHOLD,

            use_autoencoders=USE_AUTOENCODERS,

            feature_surv_analysis=FEATURE_SURV_ANALYSIS,

            cluster_method=CLUSTER_METHOD,

            cluster_eval_method=CLUSTER_EVAL_METHOD,

            classifier_type=CLASSIFIER_TYPE,

            project_name=PROJECT_NAME,

            path_results=PATH_RESULTS,

            cluster_array=CLUSTER_ARRAY,

            nb_selected_features=NB_SELECTED_FEATURES,

            mixture_params=MIXTURE_PARAMS,

            node_selection=NODES_SELECTION,

            nb_threads_coxph=NB_THREADS_COXPH,

            classification_method=CLASSIFICATION_METHOD,

            load_existing_models=LOAD_EXISTING_MODELS,

            clustering_omics=CLUSTERING_OMICS,

            path_to_save_model=PATH_TO_SAVE_MODEL,

            metadata_usage=None,

            feature_selection_usage='individual',

            use_r_packages=USE_R_PACKAGES_FOR_SURVIVAL,

            alternative_embedding=None,

            kwargs_alternative_embedding={},

            do_KM_plot=True,

            verbose=True,

            _isboosting=False,

            dataset=None,

            deep_model_additional_args={}):

        """

        """

        SimDeep.__init__(

            self,

            nb_clusters=nb_clusters,

            pvalue_thres=pvalue_thres,

            cindex_thres=cindex_thres,

            cluster_method=cluster_method,

            cluster_eval_method=cluster_eval_method,

            classifier_type=classifier_type,

            project_name=project_name,

            clustering_omics=clustering_omics,

            path_results=path_results,

            use_autoencoders=use_autoencoders,

            feature_surv_analysis=feature_surv_analysis,

            cluster_array=cluster_array,

            nb_selected_features=nb_selected_features,

            mixture_params=mixture_params,

            node_selection=node_selection,

            nb_threads_coxph=nb_threads_coxph,

            feature_selection_usage=feature_selection_usage,

            metadata_usage=metadata_usage,

            classification_method=classification_method,

            load_existing_models=load_existing_models,

            path_to_save_model=path_to_save_model,

            do_KM_plot=do_KM_plot,

            alternative_embedding=alternative_embedding,

            kwargs_alternative_embedding=kwargs_alternative_embedding,

            verbose=verbose,

            use_r_packages=use_r_packages,

            _isboosting=_isboosting,

            dataset=dataset,

            deep_model_additional_args=deep_model_additional_args)