Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenFold
Commits
d0f5531b
"examples/vscode:/vscode.git/clone" did not exist on "a48ffc523446d4594ee7c71eaa5104e9a77b84e5"
Commit
d0f5531b
authored
Nov 06, 2023
by
Christina Floristean
Browse files
Another fix for dummy msa feature creation
parent
e67d0417
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
13 additions
and
8 deletions
+13
-8
openfold/data/data_pipeline.py
openfold/data/data_pipeline.py
+13
-8
No files found.
openfold/data/data_pipeline.py
View file @
d0f5531b
...
...
@@ -272,14 +272,19 @@ def run_msa_tool(
return
result
# Generate 1-sequence MSA features having only the input sequence
def
make_dummy_msa_feats
(
input_sequence
)
->
parsers
.
Msa
:
def
make_dummy_msa_obj
(
input_sequence
)
->
parsers
.
Msa
:
deletion_matrix
=
[[
0
for
_
in
input_sequence
]]
return
parsers
.
Msa
(
sequences
=
input_sequence
,
return
parsers
.
Msa
(
sequences
=
[
input_sequence
]
,
deletion_matrix
=
deletion_matrix
,
descriptions
=
None
)
# Generate 1-sequence MSA features having only the input sequence
def
make_dummy_msa_feats
(
input_sequence
)
->
FeatureDict
:
msa_data_obj
=
make_dummy_msa_obj
(
input_sequence
)
return
make_msa_features
([
msa_data_obj
])
def
make_sequence_features_with_custom_template
(
sequence
:
str
,
mmcif_path
:
str
,
...
...
@@ -810,7 +815,7 @@ class DataPipeline:
"""
)
msa_data
[
"dummy"
]
=
make_dummy_msa_
feats
(
input_sequence
)
msa_data
[
"dummy"
]
=
make_dummy_msa_
obj
(
input_sequence
)
return
list
(
msa_data
.
values
())
...
...
@@ -885,7 +890,7 @@ class DataPipeline:
sequence_embedding_features
=
{}
# If using seqemb mode, generate a dummy MSA features using just the sequence
if
seqemb_mode
:
msa_features
=
[
make_dummy_msa_feats
(
input_sequence
)
]
msa_features
=
make_dummy_msa_feats
(
input_sequence
)
sequence_embedding_features
=
self
.
_process_seqemb_features
(
alignment_dir
)
else
:
msa_features
=
self
.
_process_msa_feats
(
alignment_dir
,
input_sequence
,
alignment_index
)
...
...
@@ -935,7 +940,7 @@ class DataPipeline:
sequence_embedding_features
=
{}
# If using seqemb mode, generate a dummy MSA features using just the sequence
if
seqemb_mode
:
msa_features
=
[
make_dummy_msa_feats
(
input_sequence
)
]
msa_features
=
make_dummy_msa_feats
(
input_sequence
)
sequence_embedding_features
=
self
.
_process_seqemb_features
(
alignment_dir
)
else
:
msa_features
=
self
.
_process_msa_feats
(
alignment_dir
,
input_sequence
,
alignment_index
)
...
...
@@ -992,7 +997,7 @@ class DataPipeline:
sequence_embedding_features
=
{}
# If in sequence embedding mode, generate dummy MSA features using just the input sequence
if
seqemb_mode
:
msa_features
=
[
make_dummy_msa_feats
(
input_sequence
)
]
msa_features
=
make_dummy_msa_feats
(
input_sequence
)
sequence_embedding_features
=
self
.
_process_seqemb_features
(
alignment_dir
)
else
:
msa_features
=
self
.
_process_msa_feats
(
alignment_dir
,
input_sequence
,
alignment_index
)
...
...
@@ -1032,7 +1037,7 @@ class DataPipeline:
sequence_embedding_features
=
{}
# If in sequence embedding mode, generate dummy MSA features using just the input sequence
if
seqemb_mode
:
msa_features
=
[
make_dummy_msa_feats
(
input_sequence
)
]
msa_features
=
make_dummy_msa_feats
(
input_sequence
)
sequence_embedding_features
=
self
.
_process_seqemb_features
(
alignment_dir
)
else
:
msa_features
=
self
.
_process_msa_feats
(
alignment_dir
,
input_sequence
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment