Skip to content

Commit 2fb66df

Browse files
committed
bugfix in model predict variable name
1 parent 3435fea commit 2fb66df

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

selene_sdk/predict/model_predict.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,7 @@ def get_predictions_for_fasta_file(self,
523523

524524

525525
def get_predictions(self,
526-
input,
526+
input_path,
527527
output_dir=None,
528528
output_format="tsv",
529529
strand_index=None):
@@ -533,7 +533,7 @@ def get_predictions(self,
533533
534534
Parameters
535535
----------
536-
input : str
536+
input_path : str
537537
A single sequence, or a path to the FASTA or BED file input.
538538
output_dir : str, optional
539539
Default is None. Output directory to write the model predictions.
@@ -579,16 +579,16 @@ def get_predictions(self,
579579
580580
"""
581581
if output_dir is None:
582-
sequence = self._pad_or_truncate_sequence(input)
582+
sequence = self._pad_or_truncate_sequence(input_path)
583583
seq_enc = self.reference_sequence.sequence_to_encoding(sequence)
584584
seq_enc = np.expand_dims(seq_enc, axis=0) # add batch size of 1
585585
return predict(self.model, seq_enc, use_cuda=self.use_cuda)
586-
elif input.endswith('.fa') or input.endswith('.fasta'):
586+
elif input_path.endswith('.fa') or input_path.endswith('.fasta'):
587587
self.get_predictions_for_fasta_file(
588-
input, output_dir, output_format=output_format)
588+
input_path, output_dir, output_format=output_format)
589589
else:
590590
self.get_predictions_for_bed_file(
591-
input,
591+
input_path,
592592
output_dir,
593593
output_format=output_format,
594594
strand_index=strand_index)

0 commit comments

Comments
 (0)