From 0611dcde73f6a62f241785fb7ca8411db2441c7d Mon Sep 17 00:00:00 2001 From: Niklas Rindtorff Date: Thu, 8 Sep 2022 12:16:30 +0200 Subject: [PATCH] revert str --- predict.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/predict.py b/predict.py index bba4ac0d..89533322 100644 --- 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)