pinder.methods package#

Subpackages#

Submodules#

pinder.methods.main module#

pinder.methods.main.json_reader(file: IO[Any]) dict[str, Any][source][source]#
pinder.methods.main.yaml_reader(file: IO[Any]) dict[str, Any][source][source]#
pinder.methods.main.preprocess(method: str, data_dir: Path, subset: str | None = None, ids: list[str] | None = None, config: dict[str, Any] = {}, pairs: SUPPORTED_PAIRS = SUPPORTED_PAIRS.ALL) None[source][source]#
pinder.methods.main.train(method: str, data_dir: Path, subset: str | None = None, ids: list[str] | None = None, config: dict[str, Any] = {}) None[source][source]#
pinder.methods.main.predict(method: str, results_dir: Path | None = None, data_dir: Path | None = None, subset: str | None = None, ids: list[str] | None = None, config: Path | None = None, pairs: SUPPORTED_PAIRS = SUPPORTED_PAIRS.ALL) None[source][source]#

Run inference/prediction with the specified method on a subset of index set or on a list of PINDER IDs.

Parameters:
  • method (str) – The method to use for prediction.

  • results_dir (Path) – Path to the directory where the results will be saved.

  • data_dir (Path) – Path to the directory where the preprocessed data is stored if available.

  • subset (str) – Optional subset of index to run prediction on (pinder-xl/s/af2).

  • ids (list) – A list of PINDER IDs to run prediction on. Takes priority over subset if provided.

  • pairs (all/holo/apo/predicted) – The type of monomer pairs to run prediction on.

  • config (Path) – An optional configuration file containing key-value parameters to pass to method. Supported formats are json and yaml.

pinder.methods.main.main() None[source][source]#

Module contents#

Namespace package root for pinder-methods.

class pinder.methods.SUPPORTED_PAIRS(value)[source][source]#

Bases: Enum

An enumeration.

ALL = 'all'#
HOLO = 'holo'#
APO = 'apo'#
PREDICTED = 'predicted'#
class pinder.methods.SUPPORTED_ACTIONS(value)[source][source]#

Bases: Enum

An enumeration.

PREPROCESS = 'preprocess'#
TRAIN = 'train'#
PREDICT = 'predict'#