predict.py as final test

labdao · Sep 8, 2022 · 6edfdb3 · 6edfdb3
1 parent 3464c36
commit 6edfdb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -105,7 +105,7 @@ def predict(
 
 if __name__ == '__main__':
     p = Predictor()
-    p.predict(protein = '/src/test/test.pdb', small_molecule_library =  '/src/test/test.sdf')
+    p.predict(protein = '/src/test/test.pdb', small_molecule_library =  '/src/test/test.txt')
 
     # print(sys.argv[0])
     # print(sys.argv[1])