Merge pull request #6 from DeepRank/rgayatri-patchtest-1

top 20 checks
DeepRank · Nov 1, 2021 · e4b69a3 · e4b69a3
2 parents 2e3939d + 51bb745
commit e4b69a3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/scripts/preprocess_bioprodict.py b/scripts/preprocess_bioprodict.py
@@ -162,8 +162,10 @@ def get_pdb_mappings(hdf5_path, pdb_root, pssm_root, variant_data):
         wt_amino_acid_code = swap[:3]
         residue_number = int(swap[3: -3])
         var_amino_acid_code = swap[-3:]
-
-        for _, row in variant_section.iterrows():  # each row maps the variant to one pdb entry
+
+        variants_section= variant_section.iloc[:20] 
+
+        for _, row in variants_section.iterrows():  # each row maps the variant to one pdb entry
 
             pdb_ac = row["pdb_structure"]
 

diff --git a/scripts/preprocess_bioprodict_with_conservation.py b/scripts/preprocess_bioprodict_with_conservation.py
@@ -193,8 +193,10 @@ def get_pdb_mappings(hdf5_path, pdb_root, variant_data):
         wt_amino_acid_code = swap[:3]
         residue_number = int(swap[3: -3])
         var_amino_acid_code = swap[-3:]
+
+        variants_section= variant_section.iloc[:20] 
 
-        for _, row in variant_section.iterrows():  # each row maps the variant to one pdb entry
+        for _, row in variants_section.iterrows():  # each row maps the variant to one pdb entry
 
             pdb_ac = row["pdb_structure"]