input fix

ntua-unit-of-control-and-informatics · Sep 10, 2024 · ba0bd21 · ba0bd21
1 parent d00d2a6
commit ba0bd21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/handlers/predict.py b/src/handlers/predict.py
@@ -49,7 +49,7 @@ def graph_post_handler(request: PredictionRequestPydantic):
     # Load the featurizer
     featurizer = SmilesGraphFeaturizer()
     featurizer.load_json_rep(feat_config)
-    smiles = request.dataset["input"][0]["SMILES"]
+    smiles = request.dataset["input"][0]
 
     def to_numpy(tensor):
         return (