From 74fc6490b5685f8145bf7dd1f3d6c5a88e066cf1 Mon Sep 17 00:00:00 2001
From: Vincent Emonet <vincent.emonet@gmail.com>
Date: Tue, 17 Oct 2023 20:23:23 +0200
Subject: [PATCH] fix predict

---
 docker-compose.yml | 2 +-
 src/predict.py     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 33005c5..299474d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -25,7 +25,7 @@ services:
     shm_size: '4g'
     # ports:
     #   - 8000:8000
-    # command: uvicorn --host 0.0.0.0 src.api:app --reload
+    command: uvicorn --host 0.0.0.0 src.api:app --reload
     networks:
       - nginx
 
diff --git a/src/predict.py b/src/predict.py
index ea2a829..e9f890a 100644
--- a/src/predict.py
+++ b/src/predict.py
@@ -72,7 +72,7 @@ def compute_drug_embedding(
         # # In this case we vectorize one by one, so only 1 row in the array
         # embeddings = vectors[0].tolist()
         # TODO: add label also?
-        embeddings = list(embed_dict[drug_smiles])
+        embeddings: np.array = embed_dict[drug_smiles].tolist()
         vector_list.append({
             "vector": embeddings,
             "payload": {