@@ -523,7 +523,7 @@ def get_predictions_for_fasta_file(self,
523
523
524
524
525
525
def get_predictions (self ,
526
- input ,
526
+ input_path ,
527
527
output_dir = None ,
528
528
output_format = "tsv" ,
529
529
strand_index = None ):
@@ -533,7 +533,7 @@ def get_predictions(self,
533
533
534
534
Parameters
535
535
----------
536
- input : str
536
+ input_path : str
537
537
A single sequence, or a path to the FASTA or BED file input.
538
538
output_dir : str, optional
539
539
Default is None. Output directory to write the model predictions.
@@ -579,16 +579,16 @@ def get_predictions(self,
579
579
580
580
"""
581
581
if output_dir is None :
582
- sequence = self ._pad_or_truncate_sequence (input )
582
+ sequence = self ._pad_or_truncate_sequence (input_path )
583
583
seq_enc = self .reference_sequence .sequence_to_encoding (sequence )
584
584
seq_enc = np .expand_dims (seq_enc , axis = 0 ) # add batch size of 1
585
585
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' ):
587
587
self .get_predictions_for_fasta_file (
588
- input , output_dir , output_format = output_format )
588
+ input_path , output_dir , output_format = output_format )
589
589
else :
590
590
self .get_predictions_for_bed_file (
591
- input ,
591
+ input_path ,
592
592
output_dir ,
593
593
output_format = output_format ,
594
594
strand_index = strand_index )
0 commit comments