Commit aacf1b6f authored by Sachin Kadyan's avatar Sachin Kadyan
Browse files

Added single seq mode in inference script and forwarded to the FASTA processors.

- Added `use_single_seq_mode` flag in inference script arguments.
- Passed on the flag to the FASTA file `data_processor`.
parent cf054ce9
...@@ -116,7 +116,9 @@ def generate_feature_dict( ...@@ -116,7 +116,9 @@ def generate_feature_dict(
local_alignment_dir = os.path.join(alignment_dir, tag) local_alignment_dir = os.path.join(alignment_dir, tag)
feature_dict = data_processor.process_fasta( feature_dict = data_processor.process_fasta(
fasta_path=tmp_fasta_path, alignment_dir=local_alignment_dir fasta_path=tmp_fasta_path,
alignment_dir=local_alignment_dir,
seqemb_mode=args.use_single_seq_mode,
) )
else: else:
with open(tmp_fasta_path, "w") as fp: with open(tmp_fasta_path, "w") as fp:
...@@ -314,6 +316,10 @@ if __name__ == "__main__": ...@@ -314,6 +316,10 @@ if __name__ == "__main__":
help="""Path to alignment directory. If provided, alignment computation help="""Path to alignment directory. If provided, alignment computation
is skipped and database path arguments are ignored.""" is skipped and database path arguments are ignored."""
) )
parser.add_argument(
"--use_single_seq_mode", action="store_true", default=False,
help="""Use single sequence embeddings instead of MSAs."""
)
parser.add_argument( parser.add_argument(
"--output_dir", type=str, default=os.getcwd(), "--output_dir", type=str, default=os.getcwd(),
help="""Name of the directory in which to output the prediction""", help="""Name of the directory in which to output the prediction""",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment