pfd.flow package
Submodules
pfd.flow.data_gen module
- class pfd.flow.data_gen.DataGen(name: str, pert_gen_op: Type[OP], prep_run_fp_op: OPTemplate, collect_data_op: Type[OP], pert_gen_step_config: dict, collect_data_config: dict, upload_python_packages: List[PathLike] | None = None)[source]
Bases:
StepsA class to represent the DataGen operation.
- name
The name of the DataGen operation.
- Type:
str
- pert_gen_op
The perturbation generation operation.
- Type:
Type[OP]
- prep_run_fp_op
The preparation and run force field operation template.
- Type:
OPTemplate
- collect_data_op
The data collection operation.
- Type:
Type[OP]
- pert_gen_step_config
Configuration for the perturbation generation step.
- Type:
dict
- collect_data_config
Configuration for the data collection step.
- Type:
dict
- upload_python_packages
List of Python packages to upload.
- Type:
Optional[List[os.PathLike]]
- property input_artifacts
Returns the input artifacts.
- Returns:
The input artifacts.
- Return type:
dict
- property input_parameters
Returns the input parameters.
- Returns:
The input parameters.
- Return type:
dict
- property output_artifacts
Returns the output artifacts.
- Returns:
The output artifacts.
- Return type:
dict
- property output_parameters
Returns the output parameters.
- Returns:
The output parameters.
- Return type:
dict
pfd.flow.distillation module
- class pfd.flow.distillation.Distillation(name: str, pert_gen_op: Type[OP], expl_dist_loop_op: OPTemplate, pert_gen_config: dict, upload_python_packages: List[PathLike] | None = None)[source]
Bases:
Steps- property input_artifacts
- property input_parameters
- property output_artifacts
- property output_parameters
pfd.flow.fine_tune module
- class pfd.flow.fine_tune.FineTune(name: str, pert_gen_op: Type[OP], prep_run_fp_op: OPTemplate, collect_data_op: Type[OP], prep_run_dp_train_op: OPTemplate, expl_finetune_loop_op: OPTemplate, pert_gen_step_config: dict, collect_data_step_config: dict, upload_python_packages: List[PathLike] | None = None, init_training=False, skip_aimd: bool = True)[source]
Bases:
Steps- property input_artifacts
- property input_parameters
- property output_artifacts
- property output_parameters