Source code for graphein.grn.config

# %%
# Graphein
# Author: Arian Jamasb <arian@jamasb.io>, Ramon Vinas
# License: MIT
# Project Website: https://github.com/a-r-j/graphein
# Code Repository: https://github.com/a-r-j/graphein

from pathlib import Path
from typing import Callable, Dict, List, Optional, Union

from pydantic import BaseModel


[docs]class TRRUSTConfig(BaseModel): """ Config object for parsing gene regulatory networks from TRRUST: https://www.grnpedia.org/trrust/ :param filtering_functions: List of functions to apply to the the TRRUST dataframe prior to graph construction. Defaults to None :type filtering_functions: List[Callable], optional :param root_dir: Specifies location of TRRUST dataset (will download to this path if not available). Defaults to None. :type root_dir: pathlib.Path, optional :param kwargs: :type kwargs: Dict[str, Union[str, int, float]], optional """ filtering_functions: Optional[List[Callable]] = None root_dir: Optional[Path] = None kwargs: Optional[Dict[str, Union[str, int, float]]] = None
[docs]class RegNetworkConfig(BaseModel): """Config object containing parameters for parsing gene regulatory networks from RegNetwork: http://regnetworkweb.org/. :param filtering_functions: List of functions to apply to the the RegNetwork dataframe prior to graph construction. Defaults to None :type filtering_functions: List[Callable], optional """ filtering_functions: Optional[List[Callable]] = None root_dir: Optional[Path] = None kwargs: Optional[Dict[str, Union[str, int, float]]] = None
[docs]class GRNGraphConfig(BaseModel): """Config object for gene regulatory network graph construction. :param kwargs: Keyword args for GRN graph construction :type kwargs: Dict[str, Union[str, int, float]], optional :param trrust_config: Config object specifying parameters for parsing TRRUST. Defaults to default config object. :type trrust_config: graphein.grn.config.TRRUSTConfig, optional :param regnetwork_config: Config object specifying parameters for parsing RegNetwork. Defaults to default config object. :type regnetwork_config: graphein.grn.config.RegNetworkConfig, optional """ kwargs: Optional[Dict[str, Union[str, int, float]]] = {} trrust_config: Optional[TRRUSTConfig] = TRRUSTConfig() regnetwork_config: Optional[RegNetworkConfig] = RegNetworkConfig() class Config: arbitrary_types_allowed: bool = True