Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -229,7 +229,9 @@ def alphafold(input_set, mode, impute):
                     new_dict = reduce_model_dict(new_dict)
                     models_we_need = list(set(itertools.chain.from_iterable(
                         [list(ov.keys()) for ok, ov in new_dict.items()])))  # Read models with these numbers
-        
+                    # Get the intersection of model we need and the models the protein has
+
+                    models_we_need = set(models_we_need).intersection(models_for_protein)
                     modelInfo = getModelInfo(uniprotID, varPos, wt, models_we_need, path_to_output_files)
         
                     for key, val in modelInfo.items():