Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -906,7 +906,7 @@ def alphafold(input_set, mode, impute):
     uniprot_matched = uniprot_matched.reset_index()
     uniprot_matched = uniprot_matched.drop(columns = {'index'})
     # Imputation
-    if (impute == 'True') or (impute == 'true'):
+    if (impute == 'True') or (impute == 'true') or (impute == True):
         filler = [20.71, 46.67, 28.13,15.5, 35.94, 21.84, 25.15, 45.15, 29.81, 29.91, 34.67, 24.72, 10.66,11.55, 13.02,
                   21.54,27.42, 38.39, 30.44, 20.9, 25.82, 46.12, 32.1, 35.96, 35.86, 37.88, 19.09, 35.2, 26.95, 37.48]
         col_index = 0