mutcleaner.cleaners.antitoxin_pard3_cleaner#

Functions

clean_antitoxin_pard3_dataset(pipeline)

Clean Antitoxin dataset using configurable pipeline

create_antitoxin_pard3_cleaner([...])

Create Antitoxin dataset cleaning pipeline

Classes

AntitoxinParD3CleanerConfig([pipeline_name, ...])

Configuration class for Antitoxin dataset cleaner.

class mutcleaner.cleaners.antitoxin_pard3_cleaner.AntitoxinParD3CleanerConfig(pipeline_name='Antitoxin Pipeline', num_workers=16, validate_config=True, column_mapping=<factory>, filters=<factory>, type_conversions=<factory>, validate_mut_workers=16, process_workers=16, label_columns=<factory>, primary_label_column='label')[source]#

Bases: BaseCleanerConfig

Configuration class for Antitoxin dataset cleaner. Inherits from BaseCleanerConfig and adds Antitoxin-specific configuration options.

Simply run mutcleaner.download_antitoxin_source_file() to download the dataset.

Alternatively, the raw Antitoxin file can be obtained from:

Attributes:
column_mappingDict[str, str]

Mapping from source to target column names

filtersDict[str, Callable]

Filter conditions for data cleaning

wt_sequencestr

Wildtype sequence for the dataset, used for mutation validation

type_conversionsDict[str, str]

Data type conversion specifications

validate_mut_workersint

Number of workers for mutation validation, set to -1 to use all available CPUs

process_workersint

Number of workers for applying mutations to sequences, set to -1 to use all available CPUs

label_columnsList[str]

List of score columns to process

primary_label_columnstr

Primary score column for the dataset

pipeline_namestr

Name of the cleaning pipeline

Methods

from_dict(config_dict)

Create configuration object from dictionary

from_json(json_path)

Load configuration from JSON file

get_summary()

Get a human-readable summary of the configuration

merge(partial_config)

Merge partial configuration with current configuration

to_dict([exclude_callables])

Convert configuration to dictionary

to_json(json_path, **json_kwargs)

Save configuration to JSON file

validate()

Validate Antitoxin-specific configuration parameters

column_mapping: Dict[str, str]#
filters: Dict[str, Callable]#
label_columns: List[str]#
pipeline_name: str = 'Antitoxin Pipeline'#
primary_label_column: str = 'label'#
process_workers: int = 16#
type_conversions: Dict[str, str]#
validate()[source]#

Validate Antitoxin-specific configuration parameters

Raises:

ValueError – If configuration is invalid

Return type:

None

validate_mut_workers: int = 16#
wt_sequence = 'MANVEKMSVAVTPQQAAVMREAVEAGEYATASEIVREAVRDWLAKRELRHDDIRRLRQLWDEGKASGRPEPVDFDALRKEARQKLTEVPPNGR'#
mutcleaner.cleaners.antitoxin_pard3_cleaner.clean_antitoxin_pard3_dataset(pipeline)[source]#

Clean Antitoxin dataset using configurable pipeline

Parameters:

pipeline (Pipeline) – Antitoxin dataset cleaning pipeline

Return type:

Tuple[Pipeline, MutationDataset]

Returns:

  • Pipeline: The cleaned pipeline - MutationDataset: The cleaned Antitoxin dataset

Examples

Use default configuration:

>>> pipeline = create_antitoxin_cleaner(df)  # df is raw Antitoxin dataset file

Use partial configuration:

>>> pipeline = create_antitoxin_cleaner(df, config={
...     "validate_mut_workers": 8,
... })

Load configuration from file:

>>> pipeline = create_antitoxin_cleaner(df, config="config.json")
>>> pipeline, dataset = clean_antitoxin_dataset(pipeline)
mutcleaner.cleaners.antitoxin_pard3_cleaner.create_antitoxin_pard3_cleaner(dataset_or_path=None, config=None)[source]#

Create Antitoxin dataset cleaning pipeline

Parameters:
  • dataset_or_path (Union[DataFrame, str, Path, None]) – Raw dataset DataFrame or file path to Antitoxin dataset.

  • config (Union[AntitoxinParD3CleanerConfig, Dict[str, Any], str, Path, None]) – Configuration for the cleaning pipeline. Can be: - AntitoxinCleanerConfig object - Dictionary with configuration parameters (merged with defaults) - Path to JSON configuration file (str or Path) - None (uses default configuration)

Return type:

Pipeline

Returns:

Pipeline: The cleaning pipeline used

Raises:
  • TypeError – If config has invalid type

  • ValueError – If configuration validation fails