def format_pdb_code_for_inputs(pdb_code: str, source_type: str):
    """Format given PDB code for prediction inputs."""
    return pdb_code[1:3] if source_type.lower() == 'input' else pdb_code.upper()