Use scorer.load()

oddt · Nov 17, 2016 · 3c1b889 · 3c1b889
1 parent 16e5987
commit 3c1b889
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bin/oddt_cli b/bin/oddt_cli
@@ -43,9 +43,9 @@ import six
 from os.path import isfile
 from ast import literal_eval
 import argparse
-import joblib
 
 import oddt
+from oddt.scoring import scorer
 
 
 def main():
@@ -189,7 +189,7 @@ def main():
 
     for score_file in args.score_file:
         if isfile(score_file):  # load pickle
-            sf = joblib.load(score_file)
+            sf = scorer.load(score_file)
             pipeline.score(sf, receptor)
         else:
             raise IOError('Could not read pickle file %s' % score_file)