in src/components/predict_relax.py [0:0]
def predict_relax(
model_features: Input[Artifact],
model_params: Input[Artifact],
prediction_runners: list,
num_ensemble: int,
run_multimer_system: bool,
is_run_relax: str,
tf_force_unified_memory: str,
xla_python_client_mem_fraction: str,
raw_predictions: Output[Artifact],
unrelaxed_proteins: Output[Artifact],
relaxed_proteins: Output[Artifact],