revert str

labdao · Sep 8, 2022 · 0611dcd · 0611dcd
1 parent fff849b
commit 0611dcd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/predict.py b/predict.py
@@ -13,8 +13,8 @@ class Predictor(BasePredictor):
     def predict(
         self,
         protein: Path = Input(description="a PDB protein structure file"),
-        small_molecule_library: str = Input(description="an SDF file containing >=2 small molecule ligands"),
-    ) -> str:    
+        small_molecule_library: Path = Input(description="an SDF file containing >=2 small molecule ligands"),
+    ) -> Path:    
         # custom changes
         args = inference_VS_2.parse_arguments()
         args = args[0]
@@ -23,7 +23,7 @@ def predict(
 
         # formatting input
         protein = str(protein)
-        #small_molecule_library = str(small_molecule_library)
+        small_molecule_library = str(small_molecule_library)
 
         # isolate the argument path basenames
         protein_base = os.path.basename(protein)