fix predict

MaastrichtU-IDS · Oct 17, 2023 · 74fc649 · 74fc649
1 parent 981835e
commit 74fc649
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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": {