in src/components/relax_protein.py [0:0]
def relax(
unrelaxed_protein: Input[Artifact],
relaxed_protein: Output[Artifact],
max_iterations: int = 0,
tolerance: float = 2.39,
stiffness: float = 10.0,
exclude_residues: List[str] = [],
max_outer_iterations: int = 3,
use_gpu: bool = True,
tf_force_unified_memory: str = '',
xla_python_client_mem_fraction: str = ''