Skip to content

Commit 97da788

Browse files
committed
Clarification in exception message for too long of sequence
1 parent 7d96114 commit 97da788

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

metapredict/backend/predictor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -564,7 +564,7 @@ def predict(inputs,
564564
# check if using gpu, specifically cuda
565565
if 'cuda' in device_string:
566566
if exceeds_max_length(inputs, max_length=MAX_CUDA_LENGTH):
567-
raise MetapredictError(f'The input sequence is too long to run on GPU. The max length for a sequence on a CUDA GPU is {MAX_CUDA_LENGTH}.\nPlease use CPU if you want to run sequences longer than 65535 amino acids.')
567+
raise MetapredictError(f'One of the input sequences is too long to run on GPU. The max length for a sequence on a CUDA GPU is {MAX_CUDA_LENGTH}.\nPlease use CPU if you want to run sequences longer than 65535 amino acids.')
568568

569569
# set device
570570
device=torch.device(device_string)

0 commit comments

Comments
 (0)