diff --git a/ImmuneBuilder/sequence_checks.py b/ImmuneBuilder/sequence_checks.py index 8c6197e..ca00a53 100644 --- a/ImmuneBuilder/sequence_checks.py +++ b/ImmuneBuilder/sequence_checks.py @@ -28,7 +28,7 @@ def number_single_sequence(sequence, chain, scheme="imgt", allowed_species=['hum # Renumber once sanity checks done if scheme == "raw": - output = [((i+1, " "),x[1]) for i,x in enumerate(numbered[0][0][0]) if x[1] != "-"] + output = [((i+1, " "),x[1]) for i,x in enumerate(output)] elif scheme != 'imgt': numbered, _, _ = anarci([("sequence", sequence)], scheme=scheme, output=False, allow=set(allow), allowed_species=allowed_species) output = [x for x in numbered[0][0][0] if x[1] != "-"]