Fix rf4

oddt · May 21, 2018 · 1c54c62 · 1c54c62
1 parent fc93a8a
commit 1c54c62
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/oddt/scoring/functions/RFScore.py b/oddt/scoring/functions/RFScore.py
@@ -115,10 +115,11 @@ def __init__(self, protein=None, n_jobs=-1, version=1, spr=0, **kwargs):
         elif version == 4:
             cutoff = 12
             mtry = 12
-            cc = close_contacts(protein,
-                                cutoff=cutoff,
-                                protein_types=protein_atomic_nums,
-                                ligand_types=ligand_atomic_nums)
+            cc = close_contacts_descriptor(
+                protein,
+                cutoff=cutoff,
+                protein_types=protein_atomic_nums,
+                ligand_types=ligand_atomic_nums)
             vina_scores = ['vina_gauss1',
                            'vina_gauss2',
                            'vina_repulsion',