def predict()

in src/components/model_predict.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

def predict(
    model_features: Input[Artifact],
    model_params: Input[Artifact],
    model_name: str,
    prediction_index: int,
    num_ensemble: int,
    run_multimer_system: bool,
    random_seed: int,
    tf_force_unified_memory: str,
    xla_python_client_mem_fraction: str,
    raw_prediction: Output[Artifact],
    unrelaxed_protein: Output[Artifact]