diff --git a/selene_sdk/predict/model_predict.py b/selene_sdk/predict/model_predict.py index e49fd0af..6be97013 100644 --- a/selene_sdk/predict/model_predict.py +++ b/selene_sdk/predict/model_predict.py @@ -460,7 +460,7 @@ def get_predictions_for_fasta_file(self, batch_ids.append([i, fasta_record.name]) sequences[i % self.batch_size, :, :] = cur_sequence_encoding - if (batch_ids and i == 0) or i % self.batch_size != 0): + if (batch_ids and i == 0) or i % self.batch_size != 0: sequences = sequences[:i % self.batch_size + 1, :, :] preds = predict(self.model, sequences, use_cuda=self.use_cuda) reporter.handle_batch_predictions(preds, batch_ids)