@@ -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