Spaces:
Sleeping
Sleeping
Commit
•
679cf0d
1
Parent(s):
82fd0af
Update code/alphafold_featureVector.py
Browse files
code/alphafold_featureVector.py
CHANGED
@@ -229,7 +229,9 @@ def alphafold(input_set, mode, impute):
|
|
229 |
new_dict = reduce_model_dict(new_dict)
|
230 |
models_we_need = list(set(itertools.chain.from_iterable(
|
231 |
[list(ov.keys()) for ok, ov in new_dict.items()]))) # Read models with these numbers
|
232 |
-
|
|
|
|
|
233 |
modelInfo = getModelInfo(uniprotID, varPos, wt, models_we_need, path_to_output_files)
|
234 |
|
235 |
for key, val in modelInfo.items():
|
|
|
229 |
new_dict = reduce_model_dict(new_dict)
|
230 |
models_we_need = list(set(itertools.chain.from_iterable(
|
231 |
[list(ov.keys()) for ok, ov in new_dict.items()]))) # Read models with these numbers
|
232 |
+
# Get the intersection of model we need and the models the protein has
|
233 |
+
|
234 |
+
models_we_need = set(models_we_need).intersection(models_for_protein)
|
235 |
modelInfo = getModelInfo(uniprotID, varPos, wt, models_we_need, path_to_output_files)
|
236 |
|
237 |
for key, val in modelInfo.items():
|